- 16 Apr, 2021 1 commit
-
-
Yezhen Cong authored
-
- 12 Apr, 2021 2 commits
-
-
twang authored
* Support nuscenes mono3d json info generation * Support nuscenes mono3d dataset class * Support attribute and bbox2d prediction in bbox3dnms and bbox3d2result * Rename dataset class and add comments to 'attrs' * Support mono3d related pipelines * Fix unittest for loading 3D annotations * Add unit test for nuscenes mono3d dataset * Rename the sample result file * Upload sample data for mono3d unit test * Upload sample data for mono3d unit test * Upload sample image for unit test * Delete tests/data/nuscenes/samples/LIDAR_TOP/CAM_BACK_LEFT directory * Add files via upload * Remove unnecessary 'f' * Remove unnecessary \ in arguments * Remove check for pycocotools version because it has been done in the cocodataset * Remove unnecessary comma, add TODO and change init of attrs in format_results * Merge RandomFlip3D and RandomFlipMono3D * Add pytest to check whether cuda is available in the unit test * Support monocular 3D detection on KITTI dataset * Add visualization TODO * Merge nus_mono3d and update dataset init * Remove duplicated loading images in mono3d * Remove aos evaluation of bbox2d predictions on KITTI * Add unit test for kitti mono3d dataset * Add accidentally deleted classes in the dataset init * Replace .format with f-string in kitti dataset * Clean comma * Toy data for unit test of kitti mono dataset * Sample image data for unit test
-
Ziyi Wu authored
* add format_results function for ScanNet test set results submission * support ScanNet test set data pre-processing * replace is_train with test_mode for better consistency * fix link error in docstring
-
- 08 Apr, 2021 1 commit
-
-
Ziyi Wu authored
* uncomplete * support lidar2img loading * add box_type_3d args in MVX-Net config file * support multi-modality demo for LiDAR point clouds * support multi-modality demo for indoor (depth) point clouds * move demo data into folder and modify docs * add input check & more general filename matching * update docs for demo and add README file for demo * add score_threshold option to demos * add data for ScanNet & KITTI dataset multi-modality test * add multi-modality visualization in ScanNet and KITTI dataset * add unittest for modified visualization function * delete saved temp file and dirs in unittests using TemporaryDirectory * fix typos in docs & move README of demos to docs/ * add demo docs to documentation * fix link error Co-authored-by:wHao-Wu <wenhaowu.chn@gmail.com>
-
- 07 Apr, 2021 3 commits
-
-
twang authored
* Support nuscenes mono3d json info generation * Support nuscenes mono3d dataset class * Support attribute and bbox2d prediction in bbox3dnms and bbox3d2result * Rename dataset class and add comments to 'attrs' * Support mono3d related pipelines * Fix unittest for loading 3D annotations * Add unit test for nuscenes mono3d dataset * Rename the sample result file * Upload sample data for mono3d unit test * Upload sample data for mono3d unit test * Upload sample image for unit test * Delete tests/data/nuscenes/samples/LIDAR_TOP/CAM_BACK_LEFT directory * Add files via upload * Remove unnecessary 'f' * Remove unnecessary \ in arguments * Remove check for pycocotools version because it has been done in the cocodataset * Remove unnecessary comma, add TODO and change init of attrs in format_results * Merge RandomFlip3D and RandomFlipMono3D * Add pytest to check whether cuda is available in the unit test * Add visualization TODO * Remove useless init in loading mono3d images
-
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.
-
Ziyi Wu authored
* remove max_num_point in ScanNet data preprocessing * add config file for ScanNet semantic segmentation dataset * modify NormalizePointsColor in pipeline * add visualization function for semantic segmentation * add ignore_index to semantic segmentation visualization function * add ignore_index to semantic segmentation evaluation function * fix ignore_index bug in semantic segmentation evaluation function * add test function to check ignore_index assignment in PointSegClassMapping * fix slicing bug in BasePoints class and add unittest * add IndoorPatchPointSample class for indoor semantic segmentation data loading and add unittest * modify LoadPointsFromFile class and its unittest to support point color loading * fix data path in unittest * add setter function for coord and attributes of BasePoint and modify unittest * modify color normalization operation to work on BasePoint class * add unittest for ScanNet semantic segmentation data loading pipeline * fix ignore_index bug in seg_eval function * add ScanNet semantic segmentation dataset and unittest * modify config file for ScanNet semantic segmentation * fix visualization function and modify unittest * fix a typo in seg_eval.py * raise exception when semantic mask is not provided in train/eval data loading * support custom computation of label weight for loss calculation * modify seg_eval function to be more efficient * fix small bugs & change variable names for clarity & add more cases to unittest * move room index resampling and label weight computation to data pre-processing * add option allowing user to determine whether to sub-sample point clouds * fix typos & change .format to f-string & fix link in comment * save all visualizations into .obj format for consistency * infer num_classes from label2cat in eval_seg function * add pre-computed room index and label weight for ScanNet dataset * replace .ply with .obj in unittests and documents * add TODO in case data is on ceph * add base dataset for all semantic segmentation tasks & add ScanNet dataset inheriting from base dataset * rename class for consistency * fix minor typos in comment * move Custom3DSegDataset to a new file * modify BasePoint setter function to enable attribute adding * add unittest for NormalizePointsColor and fix small bugs * fix unittest for BasePoints * modify ScanNet data pre-processing scripts * change ignore_idx to -1 in seg_eval function * remove sliding inference from PatchSample function and modify unittest * remove PatchSample from scannet seg test_pipeline
-
- 31 Mar, 2021 1 commit
-
-
xiliu8006 authored
* modify vote_head to support 3dssd * delete .keys() * add 3dssd unittest * 3dssd->ssd3d
-
- 28 Mar, 2021 1 commit
-
-
Ziyi Wu authored
* fix link to convert_votenet_checkpoints.py * remove redundant space in docstring * fix typos * fix link and latex citation in README * fix link error for publish_model.py * fix link error for regnet2mmdet.py * fix link error for files under tools/misc/ folder * fix link error for files under tools/analysis_tools/ folder * fix latex citation in README_zh-CN.md * fix typo and a link to mmdet
-
- 26 Mar, 2021 1 commit
-
-
xiliu8006 authored
* support nuscenes dataset in demo * add convert_SyncBN in __init__ * fix meshlab visualization bug * modify meshlab unittest * add docstring * add empty line in docstring Co-authored-by:xiliu8006 <xiliu800@gmail.com>
-
- 24 Mar, 2021 1 commit
-
-
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:ZwwWayne <wayne.zw@outlook.com>
-
- 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 2 commits
-
-
Yezhen Cong authored
* support knn gpu op * made it more robust and fixed comments
-
junhaozhang98 authored
* 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
-
- 25 Feb, 2021 2 commits
-
-
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>
-
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
-
- 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
-
- 23 Feb, 2021 1 commit
-
-
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
-
- 17 Feb, 2021 1 commit
-
-
Yezhen Cong authored
* fix bug caused by mmcv upgrade; delete pdb breakpoint * fix typos
-
- 10 Feb, 2021 2 commits
-
-
Yezhen Cong authored
-
xiliu8006 authored
* 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
-
- 01 Feb, 2021 1 commit
-
-
junhaozhang98 authored
* 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
-
- 23 Jan, 2021 1 commit
-
-
xiliu8006 authored
* 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
-
- 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>
-
- 16 Dec, 2020 1 commit
-
-
filaPro authored
* fix indoor_eval in case of less classes in prediction * fix style for formatters
-
- 28 Nov, 2020 1 commit
-
-
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 * 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
-
- 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
-
- 14 Nov, 2020 1 commit
-
-
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 points structure * modify points rotation and add coord_3d_mode unittest * modify docstring
-
- 21 Oct, 2020 1 commit
-
-
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.
-
- 20 Oct, 2020 1 commit
-
-
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
-
- 07 Oct, 2020 1 commit
-
-
twang authored
-
- 29 Sep, 2020 2 commits
-
-
encore-zhou authored
* 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
-
Wenwei Zhang authored
* fix dcn bias bug in centerpoint * fix unit tesats bug of empty iou calculation
-
- 26 Sep, 2020 1 commit
-
-
yinchimaoliang authored
* Add evaluation interval for centerpoint. * Add centerpoint unittest. * Add test unittest for centerpoint
-
- 24 Sep, 2020 1 commit
-
-
yinchimaoliang authored
* Add version assert in init file. * Fix bug in test_dcn_head.
-
- 20 Sep, 2020 1 commit
-
-
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:
wangtai <wangtai@sensetime.com> Co-authored-by:
Tai-Wang <tab_wang@outlook.com>
-
- 19 Sep, 2020 2 commits
-
-
Wenwei Zhang authored
* [Fix]: fix label type bug when using dbsampler * Unify cat_id for more general usage * fix CI bugs * keep astype np.long
-
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
-
- 15 Sep, 2020 1 commit
-
-
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 depth box3d * fix bugs of test box3d
-
- 13 Sep, 2020 1 commit
-
-
encore-zhou authored
* 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
-