1. 21 Apr, 2021 1 commit
    • Yezhen Cong's avatar
      [Fix] Fix #460 and simplify configs (#462) · 2c29afa3
      Yezhen Cong authored
      * fix #460 and simplify configs
      
      * fix duplicate key error
      
      * delete unused _delete_
      
      * add cosine docstring and fixed a bug
      
      * revert config files under benchmark folder
      
      * add type to runner in benchmark configs
      
      * remove irrelevant change
      2c29afa3
  2. 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
    • Wenhao Wu's avatar
      [Fix] Waymo loading error (#441) · 80ef7fee
      Wenhao Wu authored
      80ef7fee
  3. 14 Apr, 2021 1 commit
  4. 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
  5. 24 Mar, 2021 1 commit
    • Yezhen Cong's avatar
      [Feature] Support ImVoteNet complete model (#352) · 4eed122d
      Yezhen Cong authored
      
      
      * Added image loading in SUNRGB-D dataset (#195)
      
      * image loading
      
      * format and docstring fix
      
      * removed irrelevant files
      
      * removed irrelevant files
      
      * load image only if modality is pc+img
      
      * added modality like nuscenes
      
      * Added imvotenet image branch pretrain (#217)
      
      * image loading
      
      * naive commit
      
      * format and docstring fix
      
      * removed irrelevant files
      
      * removed irrelevant files
      
      * load image only if modality is pc+img
      
      * added modality like nuscenes
      
      * pretrain_2d_model
      
      * finetune sunrgbd
      
      * finetune sunrgbd
      
      * deleted unused code
      
      * fixed a bug
      
      * resolve conflict
      
      * update config file
      
      * fix docstring and configs
      
      * integrated vote fusion
      
      * coords transform and unit test
      
      * Update docstring
      
      * refactor and add unit test
      
      * fix bug caused by mmcv upgrade; delete pdb breakpoint
      
      * add point fusion unittest
      
      * remove unused file
      
      * fix typos
      
      * updates
      
      * add assertion info
      
      * update
      
      * add unittest
      
      * add vote fusion unittest
      
      * add vote fusion unittest
      
      * [Refactor] VoteNet refactor (#322)
      
      * votenet refactor
      
      * remove file
      
      * minor update
      
      * docstring
      
      * initial update of imvotenet
      
      * [Feature] Support vote fusion (#297)
      
      * integrated vote fusion
      
      * coords transform and unit test
      
      * Update docstring
      
      * refactor and add unit test
      
      * add point fusion unittest
      
      * remove unused file
      
      * updates
      
      * add assertion info
      
      * update
      
      * add unittest
      
      * add vote fusion unittest
      
      * add vote fusion unittest
      
      * minor update
      
      * docstring
      
      * change np ops to torch
      
      * refactor test
      
      * update
      
      * refactor of image mlp and np random ops to torch
      
      * add docstring
      
      * add config and mod dataset
      
      * fix bugs
      
      * add_comments
      
      * fix bugs
      
      * fix_bug
      
      * fix bug
      
      * fix bug
      
      * fix bug
      
      * fix bug
      
      * final fix
      
      * fix bug
      
      * ?
      
      * add docstring
      
      * move train/test cfg
      
      * change img mlp default param
      
      * rename config
      
      * minor mod
      
      * change config name
      
      * move train/test cfg
      
      * some fixes and 2d utils
      
      * fix config name
      
      * fix config override issue
      
      * config simplify & reformat
      
      * explicitly set eval mode->override train()
      
      * add fix_img_branch to config
      
      * remove set_img_branch_eval_mode
      
      * temporal fix, change calibs to calib
      
      * more docstring and view/reshape, expand/repeat change
      
      * complete imvotenet docstring
      
      * fix docstring
      
      * add config and some minor fix
      
      * rename config
      Co-authored-by: default avatarZwwWayne <wayne.zw@outlook.com>
      4eed122d
  6. 23 Feb, 2021 1 commit
    • xiliu8006's avatar
      [Enhance] Move train_cfg test_cfg to model (#307) · a481f5a8
      xiliu8006 authored
      * Move train_cfg/test_cfg to model
      
      * Move train_cfg/test_cfg to model
      
      * Move train_cfg/test_cfg to model
      
      * Move train_cfg/test_cfg to model
      
      * Move train_cfg/test_cfg to model
      
      * Move train_cfg/test_cfg to model
      
      * Move train_cfg/test_cfg to model
      
      * Move train_cfg and test_cfg into model
      
      * modify centerpoint configs
      
      * Modify docs
      
      * modify build_detector
      
      * modify test_config_build_detector
      
      * modify build_detector parameters
      
      * Adopt the same strategy in build_detector
      a481f5a8
  7. 17 Feb, 2021 1 commit
  8. 30 Nov, 2020 1 commit
  9. 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
  10. 21 Oct, 2020 3 commits
    • encore-zhou's avatar
      [Fix]: add convert script for Votenet (#163) · 64a8c455
      encore-zhou authored
      * add votenet model convert script
      
      * fix bugs of votenet config
      
      * modify scripts
      
      * add docstring
      64a8c455
    • yinchimaoliang's avatar
      [Feature]: Add double flip (#143) · ffcea26c
      yinchimaoliang authored
      * Able to double_flip test.
      
      * Remove nms_post_max_size
      
      * Able to merge then nms for tta.
      
      * Able to merge then nms for tta.
      
      * reproducible performance.
      
      * Add more configs.
      
      * Able to use scale tta.
      
      * Move scale aug to merge_aug_bboxes_3d
      
      * Remove brackets, add comments.
      
      * Delete unnecessary configs, Add instruction in README.
      
      * Add docstring.
      
      * Change readme, add empty line.
      
      * Change name.
      
      * Add tta in unittest.
      
      * Add 'to cpu'
      
      * Add simple test
      
      * Move 'to cpu'
      
      * Add full tta config
      
      * Delete circle_nms tta config, change max_num.
      
      * Update benchmark
      
      * Remove unnecessary code, move position of flip.
      
      * Complete docstring.
      
      * Add comment for double flip change.
      ffcea26c
    • twang's avatar
      [Enhance] SECOND benchmark on Waymo (#166) · f98bc952
      twang authored
      * Add second config for waymo
      
      * Update README.md
      
      * Update README.md
      
      * Fix a minor typo in the kitti log link
      f98bc952
  11. 20 Sep, 2020 1 commit
    • Wenwei Zhang's avatar
      Bump to V0.6.0 (#118) · 579b0799
      Wenwei Zhang authored
      
      
      * Add gitlab CI back
      
      * clean isort
      
      * Update gitlab CI version
      
      * Update mmcv install
      
      * fix unit test bug
      
      * waymo
      
      * Use new flake8
      
      * Update mmdet3d/core/evaluation/waymo_utils/prediction_kitti_to_waymo.py, tools/data_converter/waymo_converter.py files
      
      * Add baseline configs for waymo
      
      * fix linting
      
      * yapf reformat
      
      * update waymo results
      
      * Update waymo model zoo and docs
      
      * Bump v0.6.0
      
      * Fix a minor bug when converting waymo data
      
      * Fix cmds in the waymo doc
      
      * Fix setup.cfg to pass isort test
      
      * Fix waymo configs
      
      * Update model zoo link & doc
      
      * update version date
      
      * clean ci
      Co-authored-by: default avatarwangtai <wangtai@sensetime.com>
      Co-authored-by: default avatarTai-Wang <tab_wang@outlook.com>
      579b0799
  12. 19 Sep, 2020 1 commit
    • yinchimaoliang's avatar
      [feature]: Add Centerpoint configs (#92) · ee801168
      yinchimaoliang authored
      * Add pointpillar configs.
      
      * Add 0075 voxel configs.
      
      * Add configs, change to pipeline.
      
      * Add centerpoint.
      
      * Add 01 configs.
      
      * Change sparse shape
      
      * Change model config, change centerpoint detector to inherite from mvx two stage.
      
      * Change names of configs.
      
      * Reduce configs.
      
      * Fix pillar config can't use bug.
      
      * Add readme for centerpoint.
      
      * Streamline the code in configs.
      
      * Change readme, add nus-3d in base config
      
      * Use seperate_head config to build seperate head, remove '[0]' in simple_test_pts
      
      * Remove input_modality
      
      * Change coder weights for configs.
      
      * Change 0075 config, change order of modules in config.
      
      * Change loss_bbox config, remove unnecessary things.
      
      * Add cyclic_20e.py
      
      * Rename centerhead, move centerhead,
      
      * Change cyclic_20e
      ee801168
  13. 15 Sep, 2020 1 commit
    • encore-zhou's avatar
      [fix]: fix bug of H3DNet config (#105) · 076b2cbd
      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
      
      * fix depth box3d
      
      * fix bugs of test box3d
      076b2cbd
  14. 13 Sep, 2020 2 commits
    • encore-zhou's avatar
      [fix]:3dssd-config (#100) · fbbdd0e3
      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
      
      * fix bugs for 3dssd config
      fbbdd0e3
    • encore-zhou's avatar
      [feature]: support ssd_3d_head in 3DSSD (#83) · 460f6b3b
      encore-zhou authored
      * add ssd3dhead
      
      * fix bugs for anchorfreebboxcoder
      
      * modify ssd 3d head
      
      * modify ssd 3d head
      
      * reconstruct ssd3dhead and votehead
      
      * add unittest
      
      * modify 3dssd config
      
      * modify 3dssd head
      
      * modify 3dssd head
      
      * rename base conv bbox head
      
      * modify vote module
      
      * modify 3dssd config
      
      * fix bugs for unittest
      
      * modify test_heads.py
      
      * fix bugs for h3d bbox head
      
      * add 3dssd detector
      
      * fix bugs for 3dssd config
      
      * modify base conv bbox head
      
      * modify base conv bbox head
      
      * modify base conv bbox head
      460f6b3b
  15. 07 Sep, 2020 1 commit
  16. 01 Sep, 2020 1 commit
    • encore-zhou's avatar
      [Feature] Add pointnet2 msg and refactor pointnets (#82) · 84efe00e
      encore-zhou authored
      
      
      * add op fps with distance
      
      * add op fps with distance
      
      * modify F-DFS unittest
      
      * modify sa module
      
      * modify sa module
      
      * SA Module support D-FPS and F-FPS
      
      * modify sa module
      
      * update points sa module
      
      * modify point_sa_module
      
      * modify point sa module
      
      * reconstruct FPS
      
      * reconstruct FPS
      
      * modify docstring
      
      * add pointnet2-sa-msg backbone
      
      * modify pointnet2_sa_msg
      
      * fix merge conflicts
      
      * format tests/test_backbones.py
      
      * [Refactor]: Add registry for PointNet2Modules
      
      * modify h3dnet for base pointnet
      
      * fix docstring tweaks
      
      * fix bugs for config unittest
      Co-authored-by: default avatarZwwWayne <wayne.zw@outlook.com>
      84efe00e
  17. 30 Aug, 2020 1 commit
    • encore-zhou's avatar
      add h3d head (#58) · 16e8d143
      encore-zhou authored
      * add h3d head
      
      * add h3d roi head
      
      * update docstring of h3d roi head
      
      * reconstruct h3d head
      
      * remove unused code
      
      * modify h3d bbox head
      
      * add h3dnet init files
      
      * modify h3d bbox head
      
      * add depth_box3d unittest
      
      * update h3d head
      
      * add h3dnet benchmark
      
      * update docstring in vote_head
      
      * resovle primitive conflict
      16e8d143
  18. 03 Jul, 2020 1 commit
  19. 29 Jun, 2020 1 commit
  20. 28 Jun, 2020 1 commit
  21. 20 Jun, 2020 1 commit
  22. 18 Jun, 2020 1 commit
  23. 17 Jun, 2020 2 commits