• Sun Jiahao's avatar
    [Feature] Add MinkUNet segmentor (#2294) · ee6cc047
    Sun Jiahao authored
    * add cylindrical voxelization & voxel feature encoder
    
    * add cylindrical voxelization & voxel feature encoder
    
    * add voxel-wise label & voxelization UT
    
    * fix vfe
    
    * fix vfe UT
    
    * rename voxel encoder & add more test case
    
    * fix type hint
    
    * temporarily refactoring mmcv's voxelize and dynamic in mmdet3d for data_preprocesser
    
    * _forward
    
    * del checkpoints
    
    * add if tp
    
    * add predict
    
    * fix vfe init bug & fix UT
    
    * add grid_size & move voxelization code
    
    * fix import bug
    
    * keep radian to follow origin
    
    * add doc string
    
    * fix type hint
    
    * add minkunet voxelization and loss function
    
    * fix data
    
    * init train
    
    * fix sparsetensor typehint
    
    * rename dir
    
    * fix data config
    
    * fix data config
    
    * fix batch_size & replace dynamic_scatter
    
    * fix conflicts 2
    
    * fix conflicts on s_70
    
    * Alignment of the original implementation
    
    * rename config
    
    * add worker_init_fn_hook
    
    * remove test_config & worker hook
    
    * add UT
    
    * fix polarmix UT
    
    * add seed for cr0p5
    
    * format
    
    * rename SemanticKittiDataset
    
    * add platte & fix visual bug
    
    * add platte & fix data info bug
    
    * fix ut
    
    * fix semantic_kitti ut
    
    * fix docstring
    
    * fix config name
    
    * rename layer
    
    * fix doc string
    
    * fix review
    
    * remove filter data
    
    * fix coors typo
    
    * fix ut
    
    * pred in segmentor
    
    * fix get voxel seg
    
    * resolve comments
    ee6cc047
data_preprocessor.py 22.5 KB