-
Notifications
You must be signed in to change notification settings - Fork 7
Network Training
Joey Wilson edited this page Jun 24, 2022
·
3 revisions
-
- This directory contains TensorBoard log files for the training and validation of each model.
- Models were chosen by performance on the mean intersection over union (mIoU) metric of the validation set.
- Training was generally 24 to 48 hours, but stopped early if the model began to overfit the data.
-
- Contains the best weights for each model, determined by performance on validation set mIoU.
-
- Script for training models on CarlaSC dataset.
- Parameters are set at the top, including which network to train, where the data is located, and whether to remap the classes from 23 to 11.
- Logs training data, and saves frozen weights each epoch.
-
- Script for training models on KITTI dataset.
- Operates similarly to training file cor CarlaSC.
-
- Data loader for the CarlaSC data set.
- Loads data into raw point clouds or voxel grids, and applies data transformations if any. Also remaps semantic labels if required.
-
- General utilities for training or testing.
- Contains functions to visualize predictions, load models or resample free space for SSCNet training.
-
- Script for testing models on CarlaSC dataset.
- Parameters are set at the top, including which network to train, where the data is located, and whether to remap the classes from 23 to 11.
- Boolean flags are used to set which test to run.
-
- Script for testing models on KITTI dataset.
- Saves results in SemanticKITTI format for submission.