1. 07 Apr, 2021 1 commit
    • Ziyi Wu's avatar
      [Feature] Support ScanNet semantic segmentation dataset (#390) · 9cb75e7d
      Ziyi Wu authored
      * remove max_num_point in ScanNet data preprocessing
      
      * add config file for ScanNet semantic segmentation dataset
      
      * modify NormalizePointsColor in pipeline
      
      * add visualization function for semantic segmentation
      
      * add ignore_index to semantic segmentation visualization function
      
      * add ignore_index to semantic segmentation evaluation function
      
      * fix ignore_index bug in semantic segmentation evaluation function
      
      * add test function to check ignore_index assignment in PointSegClassMapping
      
      * fix slicing bug in BasePoints class and add unittest
      
      * add IndoorPatchPointSample class for indoor semantic segmentation data loading and add unittest
      
      * modify LoadPointsFromFile class and its unittest to support point color loading
      
      * fix data path in unittest
      
      * add setter function for coord and attributes of BasePoint and modify unittest
      
      * modify color normalization operation to work on BasePoint class
      
      * add unittest for ScanNet semantic segmentation data loading pipeline
      
      * fix ignore_index bug in seg_eval function
      
      * add ScanNet semantic segmentation dataset and unittest
      
      * modify config file for ScanNet semantic segmentation
      
      * fix visualization function and modify unittest
      
      * fix a typo in seg_eval.py
      
      * raise exception when semantic mask is not provided in train/eval data loading
      
      * support custom computation of label weight for loss calculation
      
      * modify seg_eval function to be more efficient
      
      * fix small bugs & change variable names for clarity & add more cases to unittest
      
      * move room index resampling and label weight computation to data pre-processing
      
      * add option allowing user to determine whether to sub-sample point clouds
      
      * fix typos & change .format to f-string & fix link in comment
      
      * save all visualizations into .obj format for consistency
      
      * infer num_classes from label2cat in eval_seg function
      
      * add pre-computed room index and label weight for ScanNet dataset
      
      * replace .ply with .obj in unittests and documents
      
      * add TODO in case data is on ceph
      
      * add base dataset for all semantic segmentation tasks & add ScanNet dataset inheriting from base dataset
      
      * rename class for consistency
      
      * fix minor typos in comment
      
      * move Custom3DSegDataset to a new file
      
      * modify BasePoint setter function to enable attribute adding
      
      * add unittest for NormalizePointsColor and fix small bugs
      
      * fix unittest for BasePoints
      
      * modify ScanNet data pre-processing scripts
      
      * change ignore_idx to -1 in seg_eval function
      
      * remove sliding inference from PatchSample function and modify unittest
      
      * remove PatchSample from scannet seg test_pipeline
      9cb75e7d
  2. 25 Feb, 2021 1 commit
    • encore-zhou's avatar
      [Feature] Support open3d visualization (#284) · 8214a977
      encore-zhou authored
      * add h3d backbone
      
      * add h3d backbone
      
      * add h3dnet
      
      * modify scannet config
      
      * fix bugs for proposal refine
      
      * fix bugs for test backbone
      
      * add primitive head test
      
      * modify h3dhead
      
      * modify h3d head
      
      * update loss weight config
      
      * fix bugs for h3d head loss
      
      * modify h3d head get targets function
      
      * update h3dnet base config
      
      * modify weighted loss
      
      * Revert "Merge branch 'h3d_u2' into 'master'"
      
      This reverts merge request !5
      
      * add open3d visual
      
      * modify docstring
      
      * use bbox and coord convert
      
      * modify docstring
      
      * fix bugs for unittest
      
      * fix bugs for unittest
      
      * modify doc
      
      * add visualize script and modify docs
      
      * modify docs
      
      * fix bugs
      8214a977
  3. 10 Feb, 2021 1 commit
    • xiliu8006's avatar
      [Enhance]Unittest reconstruction (#303) · 7f9b186e
      xiliu8006 authored
      * Adjust the directory structure of tests
      
      * Adopted the new directory via _get_config_directory
      
      * Adopted the new directory via _get_config_directory
      
      * Adopted the new directory via _get_config_directory
      
      * modified the wrong code comments
      7f9b186e
  4. 28 Nov, 2020 1 commit
    • encore-zhou's avatar
      [Enhance] Use Points structure in augmentation and models (#204) · 23768cba
      encore-zhou authored
      * add h3d backbone
      
      * add h3d backbone
      
      * add h3dnet
      
      * modify scannet config
      
      * fix bugs for proposal refine
      
      * fix bugs for test backbone
      
      * add primitive head test
      
      * modify h3dhead
      
      * modify h3d head
      
      * update loss weight config
      
      * fix bugs for h3d head loss
      
      * modify h3d head get targets function
      
      * update h3dnet base config
      
      * modify weighted loss
      
      * Revert "Merge branch 'h3d_u2' into 'master'"
      
      This reverts merge request !5
      
      * modify pipeline
      
      * modify kitti pipeline
      
      * fix bugs for points rotation
      
      * modify multi sweeps
      
      * modify multi sweep points
      
      * fix bugs for points slicing
      
      * modify BackgroundPointsFilter
      
      * modify pipeline
      
      * modify unittest
      
      * modify unittest
      
      * modify docstring
      
      * modify config files
      
      * update configs
      
      * modify docstring
      23768cba
  5. 22 Jul, 2020 1 commit
    • yinchimaoliang's avatar
      Add datasets unittest (#24) · 59fc5e73
      yinchimaoliang authored
      * add test_getitem for kitti
      
      * Change dataset to self
      
      * Fix dbinfo path bug
      
      * Change to 230
      
      * Change bboxes and scores
      
      * Debug, check points shape after dbsample
      
      * Debug, add assert gt_bbox and db bbox
      
      * Debug, add assert sampled and gt_bboxes
      
      * Debug, add assertion for dbinfo and gt_box
      
      * Fix transfom_3d unittest
      
      * Fix kitti dataset unittest
      
      * Add kitti data
      
      * Clean debug
      
      * Change data_augment_utils
      
      * Reduce point file
      
      * Change self to dataset
      
      * Change data_augment_utils to normal
      
      * Change order of CLASSES
      
      * Finish test_random_flip_3d
      
      * Add show unittest for scannet and sunrgbd datasets
      
      * Add show unittest for kitti dataset
      
      * Add test_evaluate for kitti dataset
      
      * Add pytest.skip
      
      * Add format_results unittest
      
      * Add test_show for lyft
      
      * Fix eval
      
      * Add test_time_aug unittest
      
      * Add bbox2result_kitti2d unittest
      
      * Change abs to np.isclose
      59fc5e73
  6. 12 Jul, 2020 1 commit
    • Wenwei Zhang's avatar
      Support GitHub action (#4) · 915ffe8b
      Wenwei Zhang authored
      * Support github action:
      
      * fix docs package
      
      * reduce action items
      
      * Update cuda arch
      
      * Set cuda arch
      
      * Add information
      
      * add cocotools in installation
      
      * skip gpu
      
      * remove duplicated module
      
      * Clean ci
      
      * Update docstring requirements
      915ffe8b
  7. 20 Jun, 2020 1 commit
  8. 17 Jun, 2020 1 commit
  9. 14 Jun, 2020 1 commit
  10. 01 Jun, 2020 1 commit
  11. 25 May, 2020 1 commit
  12. 19 May, 2020 1 commit
  13. 15 May, 2020 2 commits
  14. 14 May, 2020 4 commits
  15. 13 May, 2020 4 commits