1. 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
  2. 01 Jul, 2021 1 commit
    • hjin2902's avatar
      [Feature] Group-Free-3D head (#539) · 318499ac
      hjin2902 authored
      
      
      * group-free-3d head
      
      * GroupFree3DNet->VoteNet
      
      * modify docstring
      
      * bugfix: calculate pts_instance_label, decoder self/cross posembed init
      
      * support GroupFree3DNet, modify scannnet train config
      
      * support point cloud input features dim = 0
      
      * add groupfree3dnet test case
      
      * bugfix: softmax in decode boxes
      
      * support multi-stage predictions
      
      * modify GroupFree3DMultiheadAttention input parameters
      
      * refactor: support sunrgbd-based train
      
      * refactor: support sunrgbd-based train
      
      * fix parts of bug
      
      * modify multi-stage prediction
      
      * fixbug: conv_channels
      
      * bugfix: permute
      
      * bugfix: permute
      
      * bugfix: expand
      
      * fix MAX_NUM_OBJ=64
      
      * 4 gpu training, score_thr = 0.0
      
      * modify config, repeattime=1
      
      * bigfix: expand
      
      * modify: GroupFree3DMHA, build_positional_encoding
      
      * modify: GroupFree3DMHA, build_positional_encoding
      
      * bugfix: torch.nn
      
      * bugfix: mean loss
      
      * residual -> identity
      
      * fix name: DropOut -> Dropout
      
      * delete sunrgbd-based congfig
      
      * Fix: trailing whitespace
      
      * suffix -> prefix
      
      * bugfix: groupfree3d config
      Co-authored-by: default avatarjinhui <PJLAB\jinhui@shai14001019l.pjlab.org>
      318499ac
  3. 30 Jun, 2021 1 commit
    • Danila Rukhovich's avatar
      [Fix] Unify camera poses (#653) · ff62af6b
      Danila Rukhovich authored
      * refactor K and Rt to depth2img for SUN RGB-D
      
      * fix lint
      
      * update 3 tests
      
      * fix extra calib key and comments
      
      * remove calib from browse_dataset
      
      * fix cam to depth; rename return_z
      ff62af6b
  4. 16 Jun, 2021 2 commits
    • Danila Rukhovich's avatar
      [Feature] Support ImVoxelNet on KITTI (#627) · c1f6bbab
      Danila Rukhovich authored
      * first try of kitti
      
      * fix python3
      
      * imvoxelnet is ready for open-mmlab/mmdetection3d
      
      * apply pre-commit
      
      * update to ConvModule and AlignedAnchor3DGenerator
      
      * add unit tests
      
      * fix torch.Tensor in docstrings
      
      * revert anchor ranges
      c1f6bbab
    • 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
  5. 15 Jun, 2021 1 commit
  6. 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
  7. 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
  8. 08 May, 2021 1 commit
    • Ziyi Wu's avatar
      [Feature] Support PointNet++ decode head (#479) · 53e06229
      Ziyi Wu authored
      * support PN2 decode head
      
      * add mmseg dependency in github workflow
      
      * complete PN2 decode head
      
      * modify backbone pn2 to support seg task & its unit test
      
      * add unit test for PN2 decode_head
      53e06229
  9. 28 Apr, 2021 1 commit
  10. 23 Apr, 2021 1 commit
    • twang's avatar
      [Feature] Support FCOS3D head (#442) · f7356f4b
      twang authored
      * Support base mono3d dense head and anchor free mono3d head
      
      * Support FCOS3D head
      
      * Support FCOS3D baseline on nuScenes
      
      * Fix an import error caused by update of mmcv/mmdet
      
      * Change img_scale to scale_factor in the MultiScaleFlipAug in the config
      
      * Add pred_bbox2d in the params of anchor_free_mono3d_head
      
      * Add unit test for fcos3d head
      
      * Fix a minor bug when setting img_metas in the unit test
      
      * Add unit test for fcos3d detector
      
      * Simplify the logic of weights initialization
      
      * Add comments to specify the reason of cloning features
      
      * Update head config
      f7356f4b
  11. 16 Apr, 2021 1 commit
  12. 07 Apr, 2021 1 commit
  13. 31 Mar, 2021 1 commit
  14. 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
  15. 23 Mar, 2021 1 commit
  16. 17 Mar, 2021 1 commit
  17. 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
  18. 24 Feb, 2021 1 commit
  19. 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
  20. 17 Feb, 2021 1 commit
  21. 10 Feb, 2021 2 commits