1. 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
  2. 12 Jun, 2021 1 commit
  3. 10 Jun, 2021 1 commit
  4. 06 Jun, 2021 2 commits
  5. 01 Jun, 2021 1 commit
  6. 31 May, 2021 2 commits
  7. 26 May, 2021 3 commits
  8. 19 May, 2021 1 commit
  9. 18 May, 2021 1 commit
    • Ziyi Wu's avatar
      [Enhance] Benchmark PN++ on ScanNet and S3DIS (#541) · 7f3a16a3
      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
      7f3a16a3
  10. 14 May, 2021 1 commit
  11. 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
  12. 11 May, 2021 1 commit
  13. 30 Apr, 2021 1 commit
    • Ziyi Wu's avatar
      [Enhance] Dataset browse for multiple dataset types (#467) · 59164170
      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
      59164170
  14. 28 Apr, 2021 2 commits
    • Ziyi Wu's avatar
      [Refactor] Use MMCV MODEL_REGISTRY (#495) · d219867c
      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
      d219867c
    • Ziyi Wu's avatar
      [Fix] Fix small errors in seg dataset and cfgs (#487) · 22b4bb4e
      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
      22b4bb4e
  15. 27 Apr, 2021 1 commit
  16. 23 Apr, 2021 1 commit
    • Ziyi Wu's avatar
      [Feature] Support S3DIS data pre-processing and dataset class (#433) · a0090aa1
      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
      a0090aa1
  17. 19 Apr, 2021 2 commits
    • Ziyi Wu's avatar
      [Enhance] Add pipeline for data loading (#430) · 78c29c35
      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
      78c29c35
    • Ziyi Wu's avatar
      [Enhance] Pipeline function process points and masks simultaneously (#444) · d4d7af24
      Ziyi Wu authored
      * modify PointShuffle and add unittest
      
      * modify PointsRangeFilter and add unittest
      
      * fix small bugs in IndoorPointSample
      
      * fix small typos in docstring
      d4d7af24
  18. 12 Apr, 2021 2 commits
    • twang's avatar
      [Feature] Support monocular 3D detection on KITTI (#415) · e37f5d5e
      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
      e37f5d5e
    • Ziyi Wu's avatar
      [Feature] Support ScanNet test results submission to online benchmark (#418) · a03100ea
      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
      a03100ea
  19. 08 Apr, 2021 3 commits
    • Ziyi Wu's avatar
      [Feature] Support multi-modality visualization (demos and dataset show function) (#405) · e21e61e0
      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: default avatarwHao-Wu <wenhaowu.chn@gmail.com>
      e21e61e0
    • Wenhao Wu's avatar
      [Enhance] Support dataset browsing on all datasets (#367) · 825f47a4
      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: default avatarWuziyi616 <dazitu616@gmail.com>
      825f47a4
    • Ziyi Wu's avatar
      31165186
  20. 07 Apr, 2021 2 commits
    • twang's avatar
      [Feature] Support monocular 3D detection on nuScenes (#392) · ecd0d06a
      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
      ecd0d06a
    • Ziyi Wu's avatar
      [Feature] Support ScanNet semantic segmentation dataset (#390) · 9cb75e7d
      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
      9cb75e7d
  21. 30 Mar, 2021 1 commit
  22. 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
  23. 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
  24. 01 Feb, 2021 1 commit
    • junhaozhang98's avatar
      [Feature] SemanticKITTI Dataset (#287) · b0501727
      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
      b0501727
  25. 23 Jan, 2021 1 commit
    • xiliu8006's avatar
      Fix miss text (#278) · 87b05bae
      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
      87b05bae
  26. 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
  27. 23 Dec, 2020 1 commit
    • twang's avatar
      [Fix] Minor adjustments (#241) · dd74422e
      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
      dd74422e
  28. 28 Nov, 2020 1 commit
    • encore-zhou's avatar
      [Enhance] Use Points structure in augmentation and models (#204) · 23768cba
      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
      23768cba
  29. 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
  30. 01 Oct, 2020 1 commit