- 28 Apr, 2022 2 commits
-
-
VVsssssk authored
* support spconv2.0 * resolve comments * try to fix ut * fix ut * fix docs * add spconv2 check in collect.env * resolve comments * fix docs * fix comments
-
Shilong Zhang authored
* add quick install command * fix SyncBatchNorm * fix SyncBatchNorm
-
- 23 Mar, 2022 1 commit
-
-
Wenhao Wu authored
* import some ops from mmcv instead of mmdet3d * use mmcv ops in primitive_head.py * use mmcv ops in PAConv * remove ops in mmdet3d & fix some bugs * remove spconv & fix some bugs * fix voxelization unittest * remove spconv in ops/__init__.py * refine ops/__init__.py * recover sparse_block in ops/__init__ * fix parta2_bbox_head unittest * remove remaining ops * recover ops/__init__.py for bc breaking * add source of ops from mmcv * recover the unittest for voxelization * remove unittest
-
- 03 Mar, 2022 1 commit
-
-
Wenwei Zhang authored
Co-authored-by:Wenhao Wu <wenhaowu.chn@gmail.com>
-
- 23 Feb, 2022 1 commit
-
-
ChaimZhu authored
-
- 16 Feb, 2022 1 commit
-
-
maskjp authored
* fix bug to allow pointnet to train in fp16 * remove unused import * fix lint * fix lint for gather_points_cuda.cu Co-authored-by:peng <maskjp@tamu.edu>
-
- 09 Feb, 2022 1 commit
-
-
ChaimZhu authored
* update isort * update setuptools * fix * pre-commit run --all-files
-
- 07 Feb, 2022 1 commit
-
-
ChaimZhu authored
* add precommmit to check readme * update * fix typos
-
- 08 Dec, 2021 1 commit
-
-
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
-
- 20 Oct, 2021 1 commit
-
-
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:
Yezhen Cong <52420115+THU17cyz@users.noreply.github.com> Co-authored-by:
THU17cyz <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:
Yezhen Cong <52420115+THU17cyz@users.noreply.github.com> Co-authored-by:
THU17cyz <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:
Yezhen Cong <52420115+THU17cyz@users.noreply.github.com> Co-authored-by:
Xi Liu <75658786+xiliu8006@users.noreply.github.com> Co-authored-by:
THU17cyz <congyezhen71@hotmail.com> Co-authored-by:
xiliu8006 <xiliu8006@gmail.com>
-
- 13 Oct, 2021 1 commit
-
-
Tai-Wang authored
-
- 24 Sep, 2021 4 commits
-
-
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
-
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
-
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:
Yezhen Cong <52420115+THU17cyz@users.noreply.github.com> Co-authored-by:
THU17cyz <congyezhen71@hotmail.com>
-
Yezhen Cong authored
-
- 17 Sep, 2021 1 commit
-
-
zhanggefan authored
-
- 07 Sep, 2021 1 commit
-
-
zhanggefan authored
-
- 19 Aug, 2021 1 commit
-
-
dingchang authored
* Add header for files * Add header for files * Add header for files * Add header for files
-
- 05 Aug, 2021 1 commit
-
-
WRH authored
* use type long long and dynamic memory allocation * use int64_t instead of long long
-
- 28 Jul, 2021 1 commit
-
-
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
-
- 01 Jul, 2021 1 commit
-
-
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
-
- 16 Jun, 2021 1 commit
-
-
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
-
- 07 Jun, 2021 1 commit
-
-
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
-
- 26 May, 2021 1 commit
-
-
Ziyi Wu authored
* add knn_heap gpu op support * add unit test * remove old knn & rename knn_heap to knn * interface consistency
-
- 12 May, 2021 1 commit
-
-
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
-
- 28 Apr, 2021 1 commit
-
-
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
-
- 14 Apr, 2021 1 commit
-
-
gillbam authored
-
- 07 Apr, 2021 1 commit
-
-
zhanggefan authored
- fix 'invalid configuration argument' error triggered by empty point input. test cases covering similar situations are added to test_dynamic_scatter.py as well. trivial changes: - switch to using torch::unique_dim to generate reduce mapping instead of calculating it from scratch.
-
- 04 Apr, 2021 1 commit
-
-
Wenwei Zhang authored
* fix pt1.8 issues * fix pt1.8 issues
-
- 29 Mar, 2021 1 commit
-
-
Wenwei Zhang authored
* fix compilation error in pytorch 1.7 * add pt1.7 build * Update build.yml
-
- 24 Mar, 2021 2 commits
-
-
xiliu8006 authored
* Avoid 4 dims points * add assert prinf * modify print info
-
Yezhen Cong authored
-
- 23 Mar, 2021 1 commit
-
-
Yezhen Cong authored
* support knn gpu op * made it more robust and fixed comments * bugfix and add warning in docstring
-
- 17 Mar, 2021 1 commit
-
-
Yezhen Cong authored
* support knn gpu op * made it more robust and fixed comments
-
- 02 Mar, 2021 1 commit
-
-
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.
-
- 25 Feb, 2021 1 commit
-
-
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:zhanggefan <1152009@tongji.edu.cn>
-
- 24 Feb, 2021 1 commit
-
-
Yezhen Cong authored
* 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
-
- 22 Jan, 2021 1 commit
-
-
xiliu8006 authored
* 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>
-
- 19 Jan, 2021 1 commit
-
-
Eric Wiener authored
The tuple unpacked into three arguments when four were given when running VoteNet on sunrgbd
-
- 22 Nov, 2020 1 commit
-
-
Jintao Lin authored
* add model stastics * add model stastics and polish docs * add model stastics and polish docs * fix * chmod +x stat.py
-