GitLab

* 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: ZwwWayne <wayne.zw@outlook.com>

* support knn gpu op

* made it more robust and fixed comments

* bugfix and add warning in docstring

* support knn gpu op

* made it more robust and fixed comments

* 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

* 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: zhanggefan <1152009@tongji.edu.cn>

* 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

* fix bug caused by mmcv upgrade; delete pdb breakpoint

* fix typos

* use torch.cuda api

* update unittest for points_in_bbox_gpu and points_in_boxes_batch

* Added comments for explanation

* 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

* fix bug caused by mmcv upgrade; delete pdb breakpoint

* fix typos

* Adjust the directory structure of tests

* Adopted the new directory via _get_config_directory

* Adopted the new directory via _get_config_directory

* Adopted the new directory via _get_config_directory

* modified the wrong code comments

* 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

* fixed the bug about miss text files on kitti testing

* add bbox2result_kitti unit test

* add bbox2result_kitti unit test

* add exist_submission_file unit test

* deleted debug code and convert{} to dict()

* add _get_kitti_dataset_config()

* delete print

* Fix bug when num_features != 4

* add voxelization unittest

* fixed CI without GPU

* use the numpy version to test the CUDA version
Co-authored-by: Guanghui Ren(任广辉) <sundrops.ren@gmail.com>

* fix indoor_eval in case of less classes in prediction

* fix style for formatters

* 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

* add model stastics

* add model stastics and polish docs

* add model stastics and polish docs

* fix

* chmod +x stat.py

* 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

* 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.

* 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

* 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

* fix dcn bias bug in centerpoint

* fix unit tesats bug of empty iou calculation

* Add evaluation interval for centerpoint.

* Add centerpoint unittest.

* Add test unittest for centerpoint

* Add version assert in init file.

* Fix bug in test_dcn_head.

* 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: wangtai <wangtai@sensetime.com>
Co-authored-by: Tai-Wang <tab_wang@outlook.com>

* [Fix]: fix label type bug when using dbsampler

* Unify cat_id for more general usage

* fix CI bugs

* keep astype np.long

* 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

* 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

* 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

* add dilated ball query

* fix bugs for group points

* rename max radius

* 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

* [Feature]: support using nuimages for instance segmentation

* Update performance and start benchmark

* Change default path names

* Update performance

* Update mmcv version

* Update model links

* Update model links and fix unit tests

* rephrase

* 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: ZwwWayne <wayne.zw@outlook.com>

* 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

* add test data

* update fps test data

* update fps test data

* add data_aug BackgroundPointsFilter

* modify BackgroundPointsFilter

* modify BackgroundPointsFilter

* 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

* 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

* modify docstring

* 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.