1. 31 May, 2021 1 commit
  2. 26 May, 2021 1 commit
    • Ziyi Wu's avatar
      [Feature] Support visualization and browse_dataset on nuScenes Mono-3D dataset (#542) · c81426a3
      Ziyi Wu authored
      * fix comment errors
      
      * add eval_pipeline in mono-nuscene cfg
      
      * add vis function in nuscene-mono dataset
      
      * refactor vis function to support all three mode boxes proj to img
      
      * add unit test for nuscene-mono show func
      
      * browse_dataset support nuScenes_mono
      
      * add show_results() to SingleStageMono3DDetector
      
      * support mono-3d dataset browse
      
      * support nus_mono_dataset and single_stage_mono_detector show function
      
      * update useful_tools.md docs
      
      * support mono-3d demo
      
      * add unit test
      
      * update demo docs
      
      * fix typos & remove unused comments
      
      * polish docs
      c81426a3
  3. 19 May, 2021 2 commits
  4. 12 May, 2021 1 commit
    • Ziyi Wu's avatar
      [Feature] Support PointNet++ Segmentor (#528) · 9d852f17
      Ziyi Wu authored
      * build BaseSegmentor for point sem seg
      
      * add encoder-decoder segmentor
      
      * update mmseg dependency
      
      * fix linting errors
      
      * warp predicted seg_mask in dict
      
      * add unit test
      
      * use build_model to wrap detector and segmentor
      
      * fix compatibility with mmseg
      
      * faster sliding inference
      
      * merge master
      
      * configs for training on ScanNet
      
      * fix CI errors
      
      * add comments & fix typos
      
      * hard-code class_weight into configs
      
      * fix logger bugs
      
      * update segmentor unit test
      
      * logger use mmdet3d
      
      * use eps to replace hard-coded 1e-3
      
      * add comments
      
      * replace np operation with torch code
      
      * add comments for class_weight
      
      * add comment for BaseSegmentor.simple_test
      
      * rewrite EncoderDecoder3D to avoid inheriting from mmseg
      9d852f17
  5. 11 May, 2021 1 commit
  6. 30 Apr, 2021 2 commits
  7. 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
  8. 19 Apr, 2021 1 commit
    • 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
  9. 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
  10. 08 Apr, 2021 1 commit
    • Wenhao Wu's avatar
      [Enhance] Support dataset browsing on all datasets (#367) · 825f47a4
      Wenhao Wu authored
      
      
      * browse_dataset
      
      * move visualization config from 'configs/' to 'configs/_base_'
      
      * refine
      
      * refine
      
      * adding argument option & browsing testing split only
      
      * support get_loading_pipeline & browsing training split
      
      * support get_loading_pipeline & browsing training split
      
      * support get_loading_pipeline & browsing training split
      
      * add condition of RepeatDataset & support the usage of cfg-options & refine useful_tools.md
      
      * add condition of RepeatDataset & support the usage of cfg-options & refine useful_tools.md
      
      * enable dataset browsing with empty gt & update docs/useful_tools
      
      * support all 3d dataset
      
      * fix small typos
      
      * fix small bugs
      Co-authored-by: default avatarWuziyi616 <dazitu616@gmail.com>
      825f47a4
  11. 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
  12. 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
  13. 24 Mar, 2021 2 commits
    • 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
    • Yezhen Cong's avatar
      [Fix] Fix a bug when adding ('val', 1) to workflow (#370) · 097b66ee
      Yezhen Cong authored
      * fix a bug when adding val to workflow
      
      * update
      097b66ee
  14. 10 Mar, 2021 1 commit
    • Wenhao Wu's avatar
      Refactor train.py/test.py (#343) · 8b131f52
      Wenhao Wu authored
      * refactor
      
      * refactor train/test
      
      * refactor train/test
      
      * refactor train.py/test.py
      
      * refactor train.py/test.py --- specify default of seed to 0 & save mmdet version instead of mmdet3d git hash
      
      * refactor train.py/test.py --- specify default of seed to 0 & save mmdet version instead of mmdet3d git hash
      8b131f52
  15. 05 Mar, 2021 1 commit
  16. 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
  17. 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
  18. 27 Jan, 2021 1 commit
  19. 16 Dec, 2020 1 commit
  20. 09 Dec, 2020 1 commit
  21. 01 Dec, 2020 1 commit
  22. 22 Nov, 2020 1 commit
  23. 01 Nov, 2020 1 commit
  24. 21 Oct, 2020 1 commit
  25. 24 Sep, 2020 1 commit
  26. 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
  27. 16 Sep, 2020 1 commit
  28. 11 Sep, 2020 1 commit
    • Yuefeng Wu's avatar
      [Fix]: fix multi-batch test problem (#95) · e4857216
      Yuefeng Wu authored
      * update bbox_results to support multi-batches test and fix ci problem in test_part_aggregation_ROI_head
      
      * compress test samples in test_parta2_roi_head
      
      * compress test samples
      
      * fix multi-batch bug
      
      * fix multi-batch bug
      
      * fix multi-batch bug
      
      * fix mvxnet test
      
      * fix mvxnet test
      
      * multi batch test
      
      * multi batch test
      
      * multi batch test
      e4857216
  29. 10 Sep, 2020 1 commit
  30. 07 Sep, 2020 1 commit
  31. 29 Aug, 2020 1 commit
    • yinchimaoliang's avatar
      Change nus related (#75) · 010cd4ba
      yinchimaoliang authored
      * Add bottom2gravity and balance_class option.
      
      * Add valid_flag.
      
      * Add use_valid_flag.
      
      * Change balance_class option.
      
      * Add wrapper, remove gravity and balance_class option.
      
      * Change flag, change load_infos to repeat_indices.
      
      * Remove pdb, change docstring.
      
      * Change name, remove unnecessary args.
      
      * Remove _concat_dataset.
      
      * Change name of ClassSampledDataset.
      
      * Change docstring.
      
      * Add get_cat_ids in nus_dataset, add unittest for wrapper.
      
      * Change to len(self.dataset)
      
      * Update nus data.
      010cd4ba
  32. 07 Aug, 2020 1 commit
  33. 09 Jul, 2020 1 commit
  34. 08 Jul, 2020 2 commits