simlearner3d
Getting Started
Install Simlearner3D on Linux
Setting up a virtual environment
Install source as a package
Troubleshooting
Install Simlearner3d on WSL2 with CUDA support
Setting up WSL2
Installing Anaconda
Installing Simlearner3D
Troubleshooting
Preparing data for training
Peprocessing functions
Creating Train/Val/Test split dataset and split.csv
Create a HDF5 Dataset
Getting started quickly with a toy dataset
Performing inference on new data
Transforming .ckpt model to scripted .pt models
Run inference on a pair of epipolar images
Guides
How to train new models
Setup
Quick run
Training
Testing the model
Inference
Developer’s guide
Code versionning
Tests
Continuous Integration (CI)
Continuous Delivery (CD)
Background
KNN-Interpolation to merge multiple predictions [TODO]
General design of the package
Model should be fast, performant, and practical
Subsampling is important to improve point cloud structure
Speed is of the essence
Evaluation is key to select the right approach
Package Reference
Scripts
run
simlearner3d.train
Default configuration
simlearner3d.processing
simlearner3d.processing.datamodule.hdf5
simlearner3d.processing.dataset.hdf5
simlearner3d.processing.dataset.toy_dataset
simlearner3d.processing.dataset.utils
simlearner3d.processing.transforms.compose
simlearner3d.processing.transforms.transforms
simlearner3d.processing.transforms.augmentations
simlearner3d.models
Model
simlearner3d.models.modules
MSAFF - Unet32 - Unet Attention
simlearner3d.callbacks
Submodules
simlearner3d.callbacks.comet_callbacks
simlearner3d.callbacks.finetuning_callbacks
simlearner3d.callbacks.logging_callbacks
Module contents
simlearner3d.utils
simlearner3d.utils.utils
simlearner3d
»
Index
Index
Symbols
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
L
|
M
|
O
|
P
|
Q
|
R
|
S
|
T
|
V
Symbols
**kwargs (simlearner3d.models.generic_model.Model.forward parameter)
*args (simlearner3d.models.generic_model.Model.forward parameter)
`split` (simlearner3d.processing.dataset.toy_dataset.make_toy_dataset_from_test_file parameter)
B
batch (simlearner3d.models.generic_model.Model.test_step parameter)
(simlearner3d.models.generic_model.Model.training_step parameter)
(simlearner3d.models.generic_model.Model.validation_step parameter)
batch_idx (simlearner3d.models.generic_model.Model.test_step parameter)
(simlearner3d.models.generic_model.Model.training_step parameter)
(simlearner3d.models.generic_model.Model.validation_step parameter)
C
class_name (simlearner3d.models.generic_model.get_neural_net_class parameter)
config (simlearner3d.train.train parameter)
(simlearner3d.utils.utils.extras parameter)
(simlearner3d.utils.utils.print_config parameter)
configure_optimizers() (simlearner3d.models.generic_model.Model method)
create_hdf5() (in module simlearner3d.processing.dataset.hdf5)
CustomCompose (class in simlearner3d.processing.transforms.compose)
D
dataloader_idx (simlearner3d.models.generic_model.Model.test_step parameter)
(simlearner3d.models.generic_model.Model.training_step parameter)
(simlearner3d.models.generic_model.Model.validation_step parameter)
dataset (simlearner3d.processing.datamodule.hdf5.HDF5StereoDataModule property)
define_device_from_config_param() (in module simlearner3d.utils.utils)
E
eval_time() (in module simlearner3d.utils.utils)
EXTRACT (run.TASK_NAMES attribute)
extras() (in module simlearner3d.utils.utils)
F
fields (simlearner3d.utils.utils.print_config parameter)
files (simlearner3d.processing.dataset.toy_dataset.make_toy_dataset_from_test_file parameter)
find_file_in_dir() (in module simlearner3d.processing.dataset.utils)
FINETUNE (run.TASK_NAMES attribute)
finetune_function() (simlearner3d.callbacks.finetuning_callbacks.FinetuningFreezeUnfreeze method)
FinetuningFreezeUnfreeze (class in simlearner3d.callbacks.finetuning_callbacks)
FIT (run.TASK_NAMES attribute)
forward() (simlearner3d.models.generic_model.Model method)
(simlearner3d.models.modules.msaff.MSNet method)
freeze_before_training() (simlearner3d.callbacks.finetuning_callbacks.FinetuningFreezeUnfreeze method)
G
get_comet_logger() (in module simlearner3d.callbacks.comet_callbacks)
get_logger() (in module simlearner3d.utils.utils)
get_neural_net_class() (in module simlearner3d.models.generic_model)
H
HDF5 (run.TASK_NAMES attribute)
HDF5Dataset (class in simlearner3d.processing.dataset.hdf5)
HDF5StereoDataModule (class in simlearner3d.processing.datamodule.hdf5)
I
input_data_dir (simlearner3d.processing.dataset.utils.find_file_in_dir parameter)
L
launch_extract() (in module run)
launch_hdf5() (in module run)
launch_qualify() (in module run)
launch_train() (in module run)
log_comet_cm() (in module simlearner3d.callbacks.comet_callbacks)
log_hyperparameters() (in module simlearner3d.utils.utils)
LogCode (class in simlearner3d.callbacks.comet_callbacks)
LogLogsPath (class in simlearner3d.callbacks.comet_callbacks)
M
make_toy_dataset_from_test_file() (in module simlearner3d.processing.dataset.toy_dataset)
Model (class in simlearner3d.models.generic_model)
module
run
simlearner3d.callbacks
simlearner3d.callbacks.comet_callbacks
simlearner3d.callbacks.finetuning_callbacks
simlearner3d.models.generic_model
simlearner3d.processing.datamodule.hdf5
simlearner3d.processing.dataset.hdf5
simlearner3d.processing.dataset.toy_dataset
simlearner3d.processing.dataset.utils
simlearner3d.processing.transforms.augmentations
simlearner3d.processing.transforms.compose
simlearner3d.processing.transforms.transforms
simlearner3d.train
simlearner3d.utils.utils
MSNet (class in simlearner3d.models.modules.msaff)
O
on_train_start() (simlearner3d.callbacks.comet_callbacks.LogCode method)
P
PREDICT (run.TASK_NAMES attribute)
prepare_data() (simlearner3d.processing.datamodule.hdf5.HDF5StereoDataModule method)
prepared_data_dir (simlearner3d.processing.dataset.toy_dataset.make_toy_dataset_from_test_file parameter)
print_config() (in module simlearner3d.utils.utils)
printed and in what order. (simlearner3d.utils.utils.print_config parameter)
Q
QUALIFY (run.TASK_NAMES attribute)
R
resolve (simlearner3d.utils.utils.print_config parameter)
run
module
S
samples_hdf5_paths (simlearner3d.processing.dataset.hdf5.HDF5Dataset property)
setup() (simlearner3d.callbacks.comet_callbacks.LogLogsPath method)
(simlearner3d.processing.datamodule.hdf5.HDF5StereoDataModule method)
simlearner3d.callbacks
module
simlearner3d.callbacks.comet_callbacks
module
simlearner3d.callbacks.finetuning_callbacks
module
simlearner3d.models.generic_model
module
simlearner3d.processing.datamodule.hdf5
module
simlearner3d.processing.dataset.hdf5
module
simlearner3d.processing.dataset.toy_dataset
module
simlearner3d.processing.dataset.utils
module
simlearner3d.processing.transforms.augmentations
module
simlearner3d.processing.transforms.compose
module
simlearner3d.processing.transforms.transforms
module
simlearner3d.train
module
simlearner3d.utils.utils
module
split_csv (simlearner3d.processing.dataset.toy_dataset.make_toy_dataset_from_test_file parameter)
src_las_path (simlearner3d.processing.dataset.toy_dataset.make_toy_dataset_from_test_file parameter)
standardize_channel() (simlearner3d.processing.transforms.transforms.StandardizeIntensity method)
StandardizeIntensity (class in simlearner3d.processing.transforms.transforms)
StandardizeIntensityCenterOnZero (class in simlearner3d.processing.transforms.transforms)
T
TASK_NAMES (class in run)
(class in simlearner3d.processing.dataset.toy_dataset)
TEST (run.TASK_NAMES attribute)
test_dataloader() (simlearner3d.processing.datamodule.hdf5.HDF5StereoDataModule method)
test_step() (simlearner3d.models.generic_model.Model method)
ToTensor (class in simlearner3d.processing.transforms.transforms)
train() (in module simlearner3d.train)
train_dataloader() (simlearner3d.processing.datamodule.hdf5.HDF5StereoDataModule method)
training_step() (simlearner3d.models.generic_model.Model method)
transforms (simlearner3d.processing.transforms.compose.CustomCompose parameter)
V
val_dataloader() (simlearner3d.processing.datamodule.hdf5.HDF5StereoDataModule method)
validation_step() (simlearner3d.models.generic_model.Model method)