- 30 Jun, 2021 1 commit
-
-
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
-
- 12 Jun, 2021 1 commit
-
-
Ziyi Wu authored
* enhance IndoorPatchPointSample * modify configs and unit test * add docs & comment * fix legacy bug * minor fix * keep sample_rate and add warning
-
- 10 Jun, 2021 1 commit
-
-
Yezhen Cong authored
-
- 06 Jun, 2021 2 commits
-
-
xiliu8006 authored
* fix bg_points_filter * fix 3dssd bg_point_filter bug * fix bg_filter bug in 3dssd * update performance * add model download link
-
Ziyi Wu authored
* unit test for LoadImageFromFileMono3D * fix GlobalRotScaleTrans assertion * add unit test for ObjectNameFilter * add unit test for ObjectRangeFilter
-
- 01 Jun, 2021 1 commit
-
-
Ziyi Wu authored
* remove useless label_weight args in dataset class * modify unit tests * update compatibility docs
-
- 31 May, 2021 2 commits
- 26 May, 2021 3 commits
-
-
Ziyi Wu authored
* add RandomDropPointsColor transform * add unit test * add link to PAConv
-
Yezhen Cong authored
-
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
-
- 19 May, 2021 1 commit
-
-
Ziyi Wu authored
* fix if condition to enable KITTI detectors eval on nus/lyft datasets * fix comments
-
- 18 May, 2021 1 commit
-
-
Ziyi Wu authored
* add configs for pn2 on s3dis * fix bugs * add README for PN2 segmentor * update configs file after experiments * update README * add base schedule files for seg configs * rename pn2_seg config files * update pn2 links in README * add metafile for pn2
-
- 14 May, 2021 1 commit
-
-
Ziyi Wu authored
-
- 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
-
- 11 May, 2021 1 commit
-
-
Ziyi Wu authored
-
- 30 Apr, 2021 1 commit
-
-
Ziyi Wu authored
* fix small bug in get_loading_pipeline() * adopt eval_pipeline & support seg visualization * support multi-modality vis * fix small bugs * add multi-modality args & support S3DIS vis * update docs with imgs * fix typos
-
- 28 Apr, 2021 2 commits
-
-
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
-
Ziyi Wu authored
* fix small errors in seg dataset cfgs * add box field to 3d seg dataset * fix small bug in S3DIS data generation scripts * fix small bugs in scannet data generation
-
- 27 Apr, 2021 1 commit
-
-
Ziyi Wu authored
* efficient mapping by matrix indexing * add pointsegmapping unit test * add max_cat_id args everywhere PointSegClassMapping is called in codebase * add default value for max_cat_id for safety * add assertion of max_cat_id
-
- 23 Apr, 2021 1 commit
-
-
Ziyi Wu authored
* support S3DIS data download and pre-processing (to ScanNet format) * add S3DIS data for unittest * add S3DIS semseg dataset class and unittest * add config file for S3DIS dataset * add eval_pipeline to S3DIS dataset config file * clean code for S3DIS pre-processing scripts * reformat code * fix small bugs * resolve conflicts & modify show() to use pipeline * fix small errors * polish data pre-processing code * add more comments about S3DIS dataset * fix markdown lint error
-
- 19 Apr, 2021 2 commits
-
-
Ziyi Wu authored
* reuse pipeline in scannet-det dataset * reuse pipeline in kitti dataset * reuse pipeline in lyft dataset * reuse pipeline in sunrgbd dataset * reuse pipeline in nuscenes dataset * reuse pipeline in waymo dataset * reuse pipeline in scannet-seg dataset * add eval_pipeline in configs which have custom data pipelines * move data loading via pipeline to dataset._extract_data() for clarity * use eval_pipeline in tools/misc/visualize_results.py * get_pipeline from self when no pipeline is provided * fix small bugs * fix small bugs * simplify and clear code * remove unnecessary eval_pipeline added * add comment about why we set self.test_mode=False * small fix * modify docs about config
-
Ziyi Wu authored
* modify PointShuffle and add unittest * modify PointsRangeFilter and add unittest * fix small bugs in IndoorPointSample * fix small typos in docstring
-
- 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 3 commits
-
-
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>
-
Wenhao Wu authored
* browse_dataset * move visualization config from 'configs/' to 'configs/_base_' * refine * refine * adding argument option & browsing testing split only * support get_loading_pipeline & browsing training split * support get_loading_pipeline & browsing training split * support get_loading_pipeline & browsing training split * add condition of RepeatDataset & support the usage of cfg-options & refine useful_tools.md * add condition of RepeatDataset & support the usage of cfg-options & refine useful_tools.md * enable dataset browsing with empty gt & update docs/useful_tools * support all 3d dataset * fix small typos * fix small bugs Co-authored-by:Wuziyi616 <dazitu616@gmail.com>
-
Ziyi Wu authored
-
- 07 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 * Add visualization TODO * Remove useless init in loading mono3d images
-
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
-
- 30 Mar, 2021 1 commit
-
-
Yezhen Cong authored
* sunrgbd doc update and change in image path * rephrase
-
- 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>
-
- 25 Feb, 2021 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 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
-
- 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
-
- 12 Jan, 2021 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 bugs for empty scene * modify doc * modify filter empty scene
-
- 23 Dec, 2020 1 commit
-
-
twang authored
* Fix an old bug in box_np_ops, not used or influencing results * Update a minor filter criterion in KITTI evaluation * Update the fixed criterion, not influencing KITTI results
-
- 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
-
- 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.
-
- 01 Oct, 2020 1 commit
-
-
Wenwei Zhang authored
* Add overall evaluation * Add config waymo-3d-car.py * Update README.md * Create hv_pointpillars_secfpn_sbn_2x16_2x_waymo-3d-3class.py * Create hv_pointpillars_secfpn_sbn_2x16_2x_waymo-3d-car.py * Delete waymo-3d-car.py Instead of using new dataset config, we change the load_interval in the config for specific usage. * update url links Co-authored-by:
Tai-Wang <tab_wang@outlook.com> Co-authored-by:
twang <30491025+Tai-Wang@users.noreply.github.com>
-