1. 07 Feb, 2022 1 commit
  2. 08 Dec, 2021 1 commit
    • Wenhao Wu's avatar
      [Feature] Support PointRCNN RPN and RCNN module (#1022) · 0f8181f1
      Wenhao Wu authored
      * rebase & resubmit
      
      * rename config & model
      
      * fix unittest
      
      * resolve comments & add docstring for class_agnostic_nms
      
      * refine loss calculation & remove find_unused_parameters
      
      * resolve typo & add docstring
      
      * resolve comments
      0f8181f1
  3. 20 Oct, 2021 1 commit
    • Wenhao Wu's avatar
      [Feature] Support PointRCNN backbone (#974) · b8856a11
      Wenhao Wu authored
      
      
      * [Refactor] Main code modification for coordinate system refactor (#677)
      
      * [Enhance] Add script for data update (#774)
      
      * Fixed wrong config paths and fixed a bug in test
      
      * Fixed metafile
      
      * Coord sys refactor (main code)
      
      * Update test_waymo_dataset.py
      
      * Manually resolve conflict
      
      * Removed unused lines and fixed imports
      
      * remove coord2box and box2coord
      
      * update dir_limit_offset
      
      * Some minor improvements
      
      * Removed some \s in comments
      
      * Revert a change
      
      * Change Box3DMode to Coord3DMode where points are converted
      
      * Fix points_in_bbox function
      
      * Fix Imvoxelnet config
      
      * Revert adding a line
      
      * Fix rotation bug when batch size is 0
      
      * Keep sign of dir_scores as before
      
      * Fix several comments
      
      * Add a comment
      
      * Fix docstring
      
      * Add data update scripts
      
      * Fix comments
      
      * fix import (#839)
      
      * [Enhance]  refactor  iou_neg_piecewise_sampler.py (#842)
      
      * [Refactor] Main code modification for coordinate system refactor (#677)
      
      * [Enhance] Add script for data update (#774)
      
      * Fixed wrong config paths and fixed a bug in test
      
      * Fixed metafile
      
      * Coord sys refactor (main code)
      
      * Update test_waymo_dataset.py
      
      * Manually resolve conflict
      
      * Removed unused lines and fixed imports
      
      * remove coord2box and box2coord
      
      * update dir_limit_offset
      
      * Some minor improvements
      
      * Removed some \s in comments
      
      * Revert a change
      
      * Change Box3DMode to Coord3DMode where points are converted
      
      * Fix points_in_bbox function
      
      * Fix Imvoxelnet config
      
      * Revert adding a line
      
      * Fix rotation bug when batch size is 0
      
      * Keep sign of dir_scores as before
      
      * Fix several comments
      
      * Add a comment
      
      * Fix docstring
      
      * Add data update scripts
      
      * Fix comments
      
      * fix import
      
      * refactor iou_neg_piecewise_sampler.py
      
      * add docstring
      
      * modify docstring
      Co-authored-by: default avatarYezhen Cong <52420115+THU17cyz@users.noreply.github.com>
      Co-authored-by: default avatarTHU17cyz <congyezhen71@hotmail.com>
      
      * [Feature] Add roipooling cuda ops (#843)
      
      * [Refactor] Main code modification for coordinate system refactor (#677)
      
      * [Enhance] Add script for data update (#774)
      
      * Fixed wrong config paths and fixed a bug in test
      
      * Fixed metafile
      
      * Coord sys refactor (main code)
      
      * Update test_waymo_dataset.py
      
      * Manually resolve conflict
      
      * Removed unused lines and fixed imports
      
      * remove coord2box and box2coord
      
      * update dir_limit_offset
      
      * Some minor improvements
      
      * Removed some \s in comments
      
      * Revert a change
      
      * Change Box3DMode to Coord3DMode where points are converted
      
      * Fix points_in_bbox function
      
      * Fix Imvoxelnet config
      
      * Revert adding a line
      
      * Fix rotation bug when batch size is 0
      
      * Keep sign of dir_scores as before
      
      * Fix several comments
      
      * Add a comment
      
      * Fix docstring
      
      * Add data update scripts
      
      * Fix comments
      
      * fix import
      
      * add roipooling cuda ops
      
      * add roi extractor
      
      * add test_roi_extractor unittest
      
      * Modify setup.py to install roipooling ops
      
      * modify docstring
      
      * remove enlarge bbox in roipoint pooling
      
      * add_roipooling_ops
      
      * modify docstring
      Co-authored-by: default avatarYezhen Cong <52420115+THU17cyz@users.noreply.github.com>
      Co-authored-by: default avatarTHU17cyz <congyezhen71@hotmail.com>
      
      * [Refactor] Refactor code structure and docstrings (#803)
      
      * refactor points_in_boxes
      
      * Merge same functions of three boxes
      
      * More docstring fixes and unify x/y/z size
      
      * Add "optional" and fix "Default"
      
      * Add "optional" and fix "Default"
      
      * Add "optional" and fix "Default"
      
      * Add "optional" and fix "Default"
      
      * Add "optional" and fix "Default"
      
      * Remove None in function param type
      
      * Fix unittest
      
      * Add comments for NMS functions
      
      * Merge methods of Points
      
      * Add unittest
      
      * Add optional and default value
      
      * Fix box conversion and add unittest
      
      * Fix comments
      
      * Add unit test
      
      * Indent
      
      * Fix CI
      
      * Remove useless \\
      
      * Remove useless \\
      
      * Remove useless \\
      
      * Remove useless \\
      
      * Remove useless \\
      
      * Add unit test for box bev
      
      * More unit tests and refine docstrings in box_np_ops
      
      * Fix comment
      
      * Add deprecation warning
      
      * support pointrcnn backbone
      
      * add docstring
      
      * modify docstring
      
      * modify docstring
      
      * modify docstring
      
      * Update pointnet2_fp_neck.py
      
      * add code block
      
      * refine docstring & code
      
      * add unittest on fp_neck
      
      * refine unittest
      
      * refine unittest
      
      * refine unittest
      
      * refine unittest
      
      * refine unittest
      
      * fix docstring
      Co-authored-by: default avatarYezhen Cong <52420115+THU17cyz@users.noreply.github.com>
      Co-authored-by: default avatarXi Liu <75658786+xiliu8006@users.noreply.github.com>
      Co-authored-by: default avatarTHU17cyz <congyezhen71@hotmail.com>
      Co-authored-by: default avatarxiliu8006 <xiliu8006@gmail.com>
      b8856a11
  4. 13 Oct, 2021 1 commit
  5. 24 Sep, 2021 4 commits
    • dingchang's avatar
      [Feature] Support DGCNN (v1.0.0.dev0) (#896) · 720caedf
      dingchang authored
      * support dgcnn
      
      * support dgcnn
      
      * support dgcnn
      
      * support dgcnn
      
      * support dgcnn
      
      * support dgcnn
      
      * support dgcnn
      
      * support dgcnn
      
      * support dgcnn
      
      * support dgcnn
      
      * fix typo
      
      * fix typo
      
      * fix typo
      
      * del gf&fa registry (wo reuse pointnet module)
      
      * fix typo
      
      * add benchmark and add copyright header (for DGCNN only)
      
      * fix typo
      
      * fix typo
      
      * fix typo
      
      * fix typo
      
      * fix typo
      
      * support dgcnn
      720caedf
    • Yezhen Cong's avatar
      [Refactor] Refactor code structure and docstrings (#803) · 53435c62
      Yezhen Cong authored
      * refactor points_in_boxes
      
      * Merge same functions of three boxes
      
      * More docstring fixes and unify x/y/z size
      
      * Add "optional" and fix "Default"
      
      * Add "optional" and fix "Default"
      
      * Add "optional" and fix "Default"
      
      * Add "optional" and fix "Default"
      
      * Add "optional" and fix "Default"
      
      * Remove None in function param type
      
      * Fix unittest
      
      * Add comments for NMS functions
      
      * Merge methods of Points
      
      * Add unittest
      
      * Add optional and default value
      
      * Fix box conversion and add unittest
      
      * Fix comments
      
      * Add unit test
      
      * Indent
      
      * Fix CI
      
      * Remove useless \\
      
      * Remove useless \\
      
      * Remove useless \\
      
      * Remove useless \\
      
      * Remove useless \\
      
      * Add unit test for box bev
      
      * More unit tests and refine docstrings in box_np_ops
      
      * Fix comment
      
      * Add deprecation warning
      53435c62
    • Xi Liu's avatar
      [Feature] Add roipooling cuda ops (#843) · 4f36084f
      Xi Liu authored
      
      
      * [Refactor] Main code modification for coordinate system refactor (#677)
      
      * [Enhance] Add script for data update (#774)
      
      * Fixed wrong config paths and fixed a bug in test
      
      * Fixed metafile
      
      * Coord sys refactor (main code)
      
      * Update test_waymo_dataset.py
      
      * Manually resolve conflict
      
      * Removed unused lines and fixed imports
      
      * remove coord2box and box2coord
      
      * update dir_limit_offset
      
      * Some minor improvements
      
      * Removed some \s in comments
      
      * Revert a change
      
      * Change Box3DMode to Coord3DMode where points are converted
      
      * Fix points_in_bbox function
      
      * Fix Imvoxelnet config
      
      * Revert adding a line
      
      * Fix rotation bug when batch size is 0
      
      * Keep sign of dir_scores as before
      
      * Fix several comments
      
      * Add a comment
      
      * Fix docstring
      
      * Add data update scripts
      
      * Fix comments
      
      * fix import
      
      * add roipooling cuda ops
      
      * add roi extractor
      
      * add test_roi_extractor unittest
      
      * Modify setup.py to install roipooling ops
      
      * modify docstring
      
      * remove enlarge bbox in roipoint pooling
      
      * add_roipooling_ops
      
      * modify docstring
      Co-authored-by: default avatarYezhen Cong <52420115+THU17cyz@users.noreply.github.com>
      Co-authored-by: default avatarTHU17cyz <congyezhen71@hotmail.com>
      4f36084f
    • Yezhen Cong's avatar
  6. 17 Sep, 2021 1 commit
  7. 07 Sep, 2021 1 commit
  8. 19 Aug, 2021 1 commit
  9. 05 Aug, 2021 1 commit
  10. 28 Jul, 2021 1 commit
    • Ziyi Wu's avatar
      [Feature] Support entire PAConv and PAConvCUDA models (#783) · 3870001a
      Ziyi Wu authored
      * add PAConv decode head
      
      * add config files
      
      * add paconv's correlation loss
      
      * support reg loss in Segmentor class
      
      * minor fix
      
      * add augmentation to configs
      
      * fix ed7 in cfg
      
      * fix bug in corr loss
      
      * enable syncbn in paconv
      
      * rename to loss_regularization
      
      * rename loss_reg to loss_regularize
      
      * use SyncBN
      
      * change weight kernels to kernel weights
      
      * rename corr_loss to reg_loss
      
      * minor fix
      
      * configs fix IndoorPatchPointSample
      
      * fix grouped points minus center error
      
      * update transform_3d & add configs
      
      * merge master
      
      * fix enlarge_size bug
      
      * refine config
      
      * remove cfg files
      
      * minor fix
      
      * add comments on PAConv's ScoreNet
      
      * refine comments
      
      * update compatibility doc
      
      * remove useless lines in transforms_3d
      
      * rename with_loss_regularization to with_regularization_loss
      
      * revert palette change
      
      * remove xavier init from PAConv's ScoreNet
      3870001a
  11. 01 Jul, 2021 1 commit
    • xiliu8006's avatar
      [Refactor]: Unified parameter initialization (#622) · 07590418
      xiliu8006 authored
      * support 3dssd
      
      * support one-stage method
      
      * for lint
      
      * support two_stage
      
      * Support all methods
      
      * remove init_cfg=[] in configs
      
      * test
      
      * support h3dnet
      
      * fix lint error
      
      * fix isort
      
      * fix code style error
      
      * fix imvotenet bug
      
      * rename init_weight->init_weights
      
      * clean comma
      
      * fix test_apis does not init weights
      
      * support newest mmdet and mmcv
      
      * fix test_heads h3dnet bug
      
      * rm *.swp
      
      * remove the wrong code in build.yml
      
      * fix ssn low map
      
      * modify docs
      
      * modified ssn init_config
      
      * modify params in backbone pointnet2_sa_ssg
      
      * add ssn direction init_cfg
      
      * support segmentor
      
      * add conv a=sqrt(5)
      
      * Convmodule uses kaiming_init
      
      * fix centerpointhead init bug
      
      * add second conv2d init cfg
      
      * add unittest to confirm the input is not be modified
      
      * assert gt_bboxes_3d
      
      * rm .swag
      
      * modify docs mmdet version
      
      * adopt fcosmono3d
      
      * add fcos 3d original init method
      
      * fix mmseg version
      
      * add init cfg in fcos_mono3d.py
      
      * merge newest master
      
      * remove unused code
      
      * modify focs config due to changes of resnet
      
      * support imvoxelnet pointnet2
      
      * modified the dependencies version
      
      * support decode head
      
      * fix inference bug
      
      * modify the useless init_cfg
      
      * fix multi_modality BC-breaking
      
      * fix error blank
      
      * modify docs error
      07590418
  12. 16 Jun, 2021 1 commit
    • Ziyi Wu's avatar
      [Feature] Support PAConv modules (#599) · b07fb946
      Ziyi Wu authored
      * refactor PN SA module code
      
      * refactor PAConv op & add PAConvSAModule
      
      * add unit test
      
      * add PAConvCUDA SA module
      
      * add unit test
      
      * minor fix
      
      * merge master
      
      * move paconv sa modules to a new file
      
      * rename to BasePointSAModule
      
      * rename to PAConvCUDASAModule
      
      * delete typing comment
      
      * refine docs
      
      * rename mlp_spec to mlp_channel
      
      * fix typos in comment
      b07fb946
  13. 07 Jun, 2021 1 commit
    • Ziyi Wu's avatar
      [Feature] Support PAConv operation (#598) · c33d4ec1
      Ziyi Wu authored
      * support knn query in QueryAndGroup
      
      * add cuda implemented assign_scores op
      
      * add unit test for paconv assign_score op
      
      * refactor op
      
      * add non-cuda & cuda version PAConv
      
      * minor fix
      
      * fix cuda-9.0 compatibility
      
      * add weight init of paconv
      
      * fix typos
      
      * refactor paconv
      c33d4ec1
  14. 26 May, 2021 1 commit
  15. 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
  16. 28 Apr, 2021 1 commit
    • Ziyi Wu's avatar
      [Refactor] Use MMCV MODEL_REGISTRY (#495) · d219867c
      Ziyi Wu authored
      * remove registry.py in datasets/
      
      * remove registry.py in ops/pointnet_modules/
      
      * remove registry.py in models/
      
      * refactor builder using mmcv.MODELS registry
      
      * update mmcv min version requirement
      
      * change min mmcv version requirement in mmdet3d/__init__.py
      
      * min mmcv version to 1.3.2
      d219867c
  17. 14 Apr, 2021 1 commit
  18. 07 Apr, 2021 1 commit
  19. 04 Apr, 2021 1 commit
  20. 29 Mar, 2021 1 commit
  21. 24 Mar, 2021 2 commits
  22. 23 Mar, 2021 1 commit
  23. 17 Mar, 2021 1 commit
  24. 02 Mar, 2021 1 commit
    • zhanggefan's avatar
      [Fix] fix a bug that may cause compilation failure of dynamic voxelization... · 69047ea2
      zhanggefan authored
      [Fix] fix a bug that may cause compilation failure of dynamic voxelization when using GPUs with compute capability lower than 6.x (#326)
      
      * fix a bug that may cause compilation failure of dynamic voxelization when using gpus with compute capability lower than 6.x
      fix imperfection kernel code that may unintentionally discard valid points when input points count is larger than 50000 * 512 (nearly impossible though).
      
      * Modified scatter_points_cuda.cu to ensure backward compatibility with PyTorch1.5 on CUDA9.0
      
      * fix the issue of DynamicScatter gradient check failure by explicit mark non-floating-point tensor as non-differentiable.
      69047ea2
  25. 25 Feb, 2021 1 commit
    • zhanggefan's avatar
      A faster & more memory-efficient implementation of DynamicScatter (#318) · 93597a53
      zhanggefan authored
      
      
      * a faster & more memory-efficient implementation of DynamicScatter
      
      * fix format issues and add pytest skip code for tests on machines without cuda support
      
      * some trivial changes:
      
      decrease the number of kernel threads per block to 512, to enable inference on GPUs with computing capability lower than 2.0
      
      change the backpropagation behavior of max-reduction. when there are multiple points shares the same maximum feature value, only the first point (with lowest row index) among them is chosen to propagate the output gradient back. before this change, all points with the same maximum feature value can propagate the output gradient back. this change makes the max-reduction behaves in consistence with torch.max. this change may cause gradcheck failure in test_dynamic_scatter.py. please do not worry about it because torch.max fails the gradcheck too.
      
      * fix typo
      Co-authored-by: default avatarzhanggefan <1152009@tongji.edu.cn>
      93597a53
  26. 24 Feb, 2021 1 commit
  27. 22 Jan, 2021 1 commit
  28. 19 Jan, 2021 1 commit
  29. 22 Nov, 2020 1 commit
  30. 10 Oct, 2020 1 commit
  31. 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
  32. 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
  33. 18 Sep, 2020 2 commits
  34. 12 Sep, 2020 1 commit
  35. 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