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. 02 Apr, 2021 1 commit
  3. 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
  4. 26 Mar, 2021 1 commit
  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. 17 Mar, 2021 1 commit
    • junhaozhang98's avatar
      [Feature] Support semantic seg metrics (#332) · 573946a8
      junhaozhang98 authored
      * add ini
      
      * add semantickitti_dataset
      
      * add test semantickitti_dataset
      
      * delete last line in test_semmaticdataset
      
      * add test data
      
      * change_names
      
      * load_labels dytype
      
      * change_name
      
      * numpy
      
      * name
      
      * dtype string
      
      * minor issue-string
      
      * seg_3d_dtype
      
      * add_fast hist
      
      * add per_class iou
      
      * add seg_eval
      
      * add unitest
      
      * minor error
      
      * minor error2
      
      * minor error3
      573946a8
  7. 01 Mar, 2021 1 commit
  8. 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
  9. 22 Jan, 2021 1 commit
  10. 23 Dec, 2020 1 commit
    • twang's avatar
      [Fix] Minor adjustments (#241) · dd74422e
      twang authored
      * Fix an old bug in box_np_ops, not used or influencing results
      
      * Update a minor filter criterion in KITTI evaluation
      
      * Update the fixed criterion, not influencing KITTI results
      dd74422e
  11. 16 Dec, 2020 1 commit
  12. 01 Dec, 2020 1 commit
  13. 28 Nov, 2020 2 commits
    • Yezhen Cong's avatar
      Added axis-aligned IoU loss to VoteNet (#194) · 04cf8929
      Yezhen Cong authored
      * support axis-aligned iou loss for votenet
      
      * added doc for iou loss
      
      * fixed problems in format
      
      * updated docstring
      
      * rename and format fix
      
      * rename and format fix
      
      * rename and format fix
      
      * rename and format fix
      
      * modified config
      
      * abstracted axis_aligned_iou3d
      
      * abstracted a bbox corner decode func
      
      * fix docstring
      04cf8929
    • 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
  14. 14 Nov, 2020 1 commit
    • encore-zhou's avatar
      [Feature]: Add points structure (#196) · 20b163a4
      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 points structure
      
      * modify points rotation and add coord_3d_mode unittest
      
      * modify docstring
      20b163a4
  15. 20 Oct, 2020 1 commit
    • twang's avatar
      [Feature] Add shape-aware grouping head in SSN (#147) · b016d90c
      twang authored
      * Add shape-aware grouping head
      
      * Reformat docstrings
      
      * Remove rewritten get_anchors in shape_aware_head
      
      * Refactor and simplify shape-aware grouping head
      
      * Fix docstring
      
      * Remove fixed preset shape heads
      
      * Add unittest for AlignedAnchor3DRangeGeneratorPerCls
      
      * Add unittest for get bboxes in shape_aware_head
      
      * Add unittest for loss of shape_aware_head
      
      * Fix unstandard docstrings
      
      * Minor fix for a comment
      
      * Add assertion to make sure not all boxes are filtered
      b016d90c
  16. 07 Oct, 2020 1 commit
  17. 29 Sep, 2020 1 commit
    • encore-zhou's avatar
      [Feature]:add sweep point sample (#125) · 24abf9b0
      encore-zhou authored
      * add sweep points sampler
      
      * add sweep points sampler
      
      * rebase master
      
      * modify unittest
      
      * modify voxel based point sampler
      
      * modify voxel based point sampler
      
      * fix bugs for voxel based point sampler
      
      * modify voxel based points sampler
      
      * fix bugs
      
      * modify repr string
      24abf9b0
  18. 24 Sep, 2020 1 commit
  19. 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
  20. 19 Sep, 2020 1 commit
    • yinchimaoliang's avatar
      [feature]: Add multi-group head of CenterPoint (#49) · d07bd8bd
      yinchimaoliang authored
      * Add modules.
      
      * Add test_center_head.
      
      * Add docstring.
      
      * Change comments.
      
      * Add dcn_head.
      
      * Add doc_string.
      
      * Add get_targets.
      
      * Can use_get_targets.
      
      * get_targets results aligned.
      
      * Use box_structure.
      
      * Use get_targets_single.
      
      * Add docstring.
      
      * Fix dcn_center_head unittest.
      
      * Delete unnecessary unittest.
      
      * Add docstring.
      
      * Change format.
      
      * Add circle_nms.
      
      * Change structure of mg_head.
      
      * Add bbox coder for centerpoint.
      
      * Add docstrings.
      
      * Add docstrings.
      
      * Add get_bboxes and unittest.
      
      * Change docstring.
      
      * Add img_metas.
      
      * Change bbox coder unittest.
      
      * Add task_detections.
      
      * Change docstring.
      
      * Change circle_nms to cpu.
      
      * Change test_nms.
      
      * Change score_th, chang keep to long type.
      
      * Change docstring and unittest.
      
      * Remove unnecessary things.
      
      * Move gaussian.
      
      * move clip_sigmoid, change dict.
      
      * Change config.
      
      * Change test_heads.
      
      * Move weight initialization to init_weights func.
      
      * Remove loc_loss_element adn==nd num_postive.
      
      * Change bboxes to the right format.
      
      * Change loss and bbox order.
      
      * Update test_heads.
      
      * Change loss.
      
      * Change names in mg_head, change head unittest.
      
      * Remove centerpoint_focal_loss, change docstring.
      
      * Change topK default to 80.
      
      * Change boxes in test_nms. Change task_boxes defaults to None.
      
      * Fix rotate nms bug.
      
      * Change docstring.
      
      * Add docstring for get_task_detection and loss.
      
      * Remove gaussian funcs, change mg_head.
      
      * Change gaussianfocalloss to mean.
      
      * change centerpoint_bbox_coder '/' to torch.div, fix centerhead unittest.
      
      * Change div to '/'
      
      * Change order in centerpoint_coder, change names, change dcn layer.
      
      * Fix import in __init__
      
      * Add gaussian unittest.
      
      * Remove np ops in mg_head.
      
      * Update docstring.
      
      * Fix docstring use config to build head.
      
      * Remove **kwargs
      
      * Remove unnecessary codes, change order of bboxes.
      
      * Remove '\' in args and pdb, change loss_bbox.
      
      * Fix test_heads unittest.
      
      * Remove unnecessary comments
      
      * Change bbox order in rotate nms.
      
      * Remove unnecessary attributes
      
      * Change name, remove float
      d07bd8bd
  21. 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
  22. 13 Sep, 2020 1 commit
    • 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
  23. 01 Sep, 2020 1 commit
    • Wenwei Zhang's avatar
      Fix all warnings in pt1.6 (#72) · dde4b02c
      Wenwei Zhang authored
      * Fix (torch.nonzero): Fix  warning of torch.nonzero and bug of nms_iou
      
      * Fix contiguous warning of tensor
      
      * Update CI with pt1.6
      
      * update build action
      
      * recover ci
      
      * Add force cuda
      
      * Use mmcv-full==1.0.5
      
      * Fix nonzero warning
      
      * Update CI config
      
      * update links
      
      * fix unit tests
      
      * fix unit tests
      
      * Lock pytorch version in CI
      
      * Lock pytorch version in CI
      
      * Lock pytorch version in CI
      
      * Try specify torchlink
      
      * Try specify torchlink
      
      * Only add 3.7 for CI
      
      * fix synctax error
      
      * Use 1.3.0
      
      * Use 1.3.0 and specify mmcv
      
      * give up 1.3
      
      * fix cublas issue
      
      * fix cublas issue
      dde4b02c
  24. 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
  25. 20 Aug, 2020 1 commit
  26. 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
  27. 11 Jul, 2020 1 commit
    • Wenwei Zhang's avatar
      Refine docs and license (#11) · 5d1cbe77
      Wenwei Zhang authored
      * Refine docs and license
      
      * Change order of install
      
      * Clean unnecessary requirements
      
      * Install mmdet manually in readthedocs
      
      * delete mmdet install
      
      * reformat docstring in code
      
      * Refine docs
      
      * Reformat
      
      * fix format issue
      5d1cbe77
  28. 09 Jul, 2020 2 commits
  29. 08 Jul, 2020 4 commits
  30. 07 Jul, 2020 2 commits
  31. 06 Jul, 2020 3 commits
  32. 05 Jul, 2020 1 commit