Train a Semantic Segmentation Model Using PyTorch
github.com › train_ss_model_using_pytorch# Training Semantic Segmentation Model using PyTorch # import torch import open3d.ml.torch as ml3d # Read a dataset by specifying the path. We are also providing the cache directory and training split. dataset = ml3d.datasets.SemanticKITTI(dataset_path= ' ../datasets/ ' , cache_dir= ' ./logs/cache ' ,training_split=[ ' 00 ' , ' 01 ' , ' 02 ' , ' 03 ' , ' 04 ' , ' 05 ' , ' 06 ' , ' 07 ' , ' 09 ' , ' 10 ' ]) # Split the dataset for 'training'.