1. 30 Apr, 2021 1 commit
    • Ziyi Wu's avatar
      [Enhance] Add waymo unit test (#455) · 951b5219
      Ziyi Wu authored
      * add sample data for waymo dataset unit test
      
      * support getitem unit test
      
      * unit test for show
      
      * unfinished evaluate unit test
      
      * KITTI format evaluate done, Waymo format uncomplete
      
      * add format_results unit test
      
      * make waymo sample data size small
      
      * remove unused comment
      
      * add pipeline show test
      
      * finish
      
      * compress img size
      951b5219
  2. 27 Apr, 2021 1 commit
  3. 23 Apr, 2021 1 commit
    • Ziyi Wu's avatar
      [Feature] Support S3DIS data pre-processing and dataset class (#433) · a0090aa1
      Ziyi Wu authored
      * support S3DIS data download and pre-processing (to ScanNet format)
      
      * add S3DIS data for unittest
      
      * add S3DIS semseg dataset class and unittest
      
      * add config file for S3DIS dataset
      
      * add eval_pipeline to S3DIS dataset config file
      
      * clean code for S3DIS pre-processing scripts
      
      * reformat code
      
      * fix small bugs
      
      * resolve conflicts & modify show() to use pipeline
      
      * fix small errors
      
      * polish data pre-processing code
      
      * add more comments about S3DIS dataset
      
      * fix markdown lint error
      a0090aa1
  4. 19 Apr, 2021 2 commits
    • Ziyi Wu's avatar
      [Enhance] Add pipeline for data loading (#430) · 78c29c35
      Ziyi Wu authored
      * reuse pipeline in scannet-det dataset
      
      * reuse pipeline in kitti dataset
      
      * reuse pipeline in lyft dataset
      
      * reuse pipeline in sunrgbd dataset
      
      * reuse pipeline in nuscenes dataset
      
      * reuse pipeline in waymo dataset
      
      * reuse pipeline in scannet-seg dataset
      
      * add eval_pipeline in configs which have custom data pipelines
      
      * move data loading via pipeline to dataset._extract_data() for clarity
      
      * use eval_pipeline in tools/misc/visualize_results.py
      
      * get_pipeline from self when no pipeline is provided
      
      * fix small bugs
      
      * fix small bugs
      
      * simplify and clear code
      
      * remove unnecessary eval_pipeline added
      
      * add comment about why we set self.test_mode=False
      
      * small fix
      
      * modify docs about config
      78c29c35
    • Ziyi Wu's avatar
      [Enhance] Pipeline function process points and masks simultaneously (#444) · d4d7af24
      Ziyi Wu authored
      * modify PointShuffle and add unittest
      
      * modify PointsRangeFilter and add unittest
      
      * fix small bugs in IndoorPointSample
      
      * fix small typos in docstring
      d4d7af24
  5. 12 Apr, 2021 2 commits
    • twang's avatar
      [Feature] Support monocular 3D detection on KITTI (#415) · e37f5d5e
      twang authored
      * Support nuscenes mono3d json info generation
      
      * Support nuscenes mono3d dataset class
      
      * Support attribute and bbox2d prediction in bbox3dnms and bbox3d2result
      
      * Rename dataset class and add comments to 'attrs'
      
      * Support mono3d related pipelines
      
      * Fix unittest for loading 3D annotations
      
      * Add unit test for nuscenes mono3d dataset
      
      * Rename the sample result file
      
      * Upload sample data for mono3d unit test
      
      * Upload sample data for mono3d unit test
      
      * Upload sample image for unit test
      
      * Delete tests/data/nuscenes/samples/LIDAR_TOP/CAM_BACK_LEFT directory
      
      * Add files via upload
      
      * Remove unnecessary 'f'
      
      * Remove unnecessary \ in arguments
      
      * Remove check for pycocotools version because it has been done in the cocodataset
      
      * Remove unnecessary comma, add TODO and change init of attrs in format_results
      
      * Merge RandomFlip3D and RandomFlipMono3D
      
      * Add pytest to check whether cuda is available in the unit test
      
      * Support monocular 3D detection on KITTI dataset
      
      * Add visualization TODO
      
      * Merge nus_mono3d and update dataset init
      
      * Remove duplicated loading images in mono3d
      
      * Remove aos evaluation of bbox2d predictions on KITTI
      
      * Add unit test for kitti mono3d dataset
      
      * Add accidentally deleted classes in the dataset init
      
      * Replace .format with f-string in kitti dataset
      
      * Clean comma
      
      * Toy data for unit test of kitti mono dataset
      
      * Sample image data for unit test
      e37f5d5e
    • Ziyi Wu's avatar
      [Feature] Support ScanNet test results submission to online benchmark (#418) · a03100ea
      Ziyi Wu authored
      * add format_results function for ScanNet test set results submission
      
      * support ScanNet test set data pre-processing
      
      * replace is_train with test_mode for better consistency
      
      * fix link error in docstring
      a03100ea
  6. 08 Apr, 2021 1 commit
    • Ziyi Wu's avatar
      [Feature] Support multi-modality visualization (demos and dataset show function) (#405) · e21e61e0
      Ziyi Wu authored
      
      
      * uncomplete
      
      * support lidar2img loading
      
      * add box_type_3d args in MVX-Net config file
      
      * support multi-modality demo for LiDAR point clouds
      
      * support multi-modality demo for indoor (depth) point clouds
      
      * move demo data into folder and modify docs
      
      * add input check & more general filename matching
      
      * update docs for demo and add README file for demo
      
      * add score_threshold option to demos
      
      * add data for ScanNet & KITTI dataset multi-modality test
      
      * add multi-modality visualization in ScanNet and KITTI dataset
      
      * add unittest for modified visualization function
      
      * delete saved temp file and dirs in unittests using TemporaryDirectory
      
      * fix typos in docs & move README of demos to docs/
      
      * add demo docs to documentation
      
      * fix link error
      Co-authored-by: default avatarwHao-Wu <wenhaowu.chn@gmail.com>
      e21e61e0
  7. 07 Apr, 2021 2 commits
    • twang's avatar
      [Feature] Support monocular 3D detection on nuScenes (#392) · ecd0d06a
      twang authored
      * Support nuscenes mono3d json info generation
      
      * Support nuscenes mono3d dataset class
      
      * Support attribute and bbox2d prediction in bbox3dnms and bbox3d2result
      
      * Rename dataset class and add comments to 'attrs'
      
      * Support mono3d related pipelines
      
      * Fix unittest for loading 3D annotations
      
      * Add unit test for nuscenes mono3d dataset
      
      * Rename the sample result file
      
      * Upload sample data for mono3d unit test
      
      * Upload sample data for mono3d unit test
      
      * Upload sample image for unit test
      
      * Delete tests/data/nuscenes/samples/LIDAR_TOP/CAM_BACK_LEFT directory
      
      * Add files via upload
      
      * Remove unnecessary 'f'
      
      * Remove unnecessary \ in arguments
      
      * Remove check for pycocotools version because it has been done in the cocodataset
      
      * Remove unnecessary comma, add TODO and change init of attrs in format_results
      
      * Merge RandomFlip3D and RandomFlipMono3D
      
      * Add pytest to check whether cuda is available in the unit test
      
      * Add visualization TODO
      
      * Remove useless init in loading mono3d images
      ecd0d06a
    • 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
  8. 28 Mar, 2021 1 commit
    • Ziyi Wu's avatar
      [Fix] Fix typos and errors (#386) · dabf0a26
      Ziyi Wu authored
      * fix link to convert_votenet_checkpoints.py
      
      * remove redundant space in docstring
      
      * fix typos
      
      * fix link and latex citation in README
      
      * fix link error for publish_model.py
      
      * fix link error for regnet2mmdet.py
      
      * fix link error for files under tools/misc/ folder
      
      * fix link error for files under tools/analysis_tools/ folder
      
      * fix latex citation in README_zh-CN.md
      
      * fix typo and a link to mmdet
      dabf0a26
  9. 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
  10. 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