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. 21 Jul, 2021 2 commits
    • Tai-Wang's avatar
      [Fix] Fix the dimension and yaw hack of FCOS3D on nuScenes (#744) · aae48a96
      Tai-Wang authored
      * Fix rotation and dim hacks between nuscbox and our cambox
      
      * Remove hack in the mono browse, fix cam_intrinsic in the RandomFlip3D
      
      * Apply a more suitable hack transformation (dim & yaw hack)
      
      * Fix incorrect transformation of dim in post-processing
      
      * Remove transformation from global to local yaw in kitti mono dataset
      
      * Add comments for cam_intrinsic modification in the doc and code
      
      * Fix typos and invalid links
      
      * Add compatibility doc for fixing nus hacks
      
      * Update benchmark
      aae48a96
    • xiliu8006's avatar
      cbf194fa
  3. 20 Jul, 2021 1 commit
  4. 14 Jul, 2021 1 commit
  5. 13 Jul, 2021 2 commits
    • xiliu8006's avatar
      [Enhance] Modify the Fcos3d init_weights methods (#705) · 6a733eb4
      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
      
      * add init cfg
      
      * add init cfg
      
      * modify fcos3d init_weights
      
      * deleted error merge code
      
      * deleted useless code
      
      * remove useless code
      6a733eb4
    • Ziyi Wu's avatar
      [Feature] Support FLOPs and Params calculation in MMDet3D (#736) · 6cd23071
      Ziyi Wu authored
      * add get_flops.py
      
      * modify SSD to support flops calc
      
      * update docs
      
      * support mono3d models
      6cd23071
  6. 01 Jul, 2021 2 commits
    • 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
    • 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
  7. 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
  8. 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
  9. 15 Jun, 2021 1 commit
  10. 06 Jun, 2021 1 commit
    • xiliu8006's avatar
      [Fix] Fix bg point filter (#609) · cde515d5
      xiliu8006 authored
      * fix bg_points_filter
      
      * fix 3dssd bg_point_filter bug
      
      * fix bg_filter bug in 3dssd
      
      * update performance
      
      * add model download link
      cde515d5
  11. 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
  12. 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
  13. 10 May, 2021 1 commit
  14. 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
  15. 30 Apr, 2021 2 commits
  16. 28 Apr, 2021 2 commits
  17. 27 Apr, 2021 1 commit
  18. 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
  19. 21 Apr, 2021 1 commit
  20. 14 Apr, 2021 1 commit
    • twang's avatar
      [Feature] Support FCOS3D Detector (#436) · 4002f06c
      twang authored
      * Support single stage mono3d detector
      
      * Add single stage mono3d detector in the init
      
      * Support FCOS Mono3D detector
      
      * Minor adjustment of docstring
      
      * Fix detector import
      
      * Move numpy import to header
      4002f06c
  21. 31 Mar, 2021 1 commit
  22. 30 Mar, 2021 1 commit
  23. 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
  24. 02 Mar, 2021 1 commit
  25. 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
  26. 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
  27. 17 Feb, 2021 1 commit
  28. 10 Feb, 2021 1 commit
  29. 12 Jan, 2021 1 commit
    • encore-zhou's avatar
      [Fix] fix bugs for 3DSSD triggered by empty GT (#258) · 58985341
      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 empty scene
      
      * modify doc
      
      * modify filter empty scene
      58985341
  30. 13 Dec, 2020 1 commit
  31. 28 Nov, 2020 1 commit
    • 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
  32. 22 Nov, 2020 1 commit
  33. 21 Oct, 2020 1 commit
    • 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
  34. 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