1. 20 Apr, 2022 1 commit
  2. 16 Feb, 2022 1 commit
  3. 09 Feb, 2022 2 commits
  4. 21 Jan, 2022 1 commit
  5. 13 Dec, 2021 1 commit
    • ChaimZhu's avatar
      [Feature] Add MonoFlex data augmentation label generation function (#1026) · c6c3c46d
      ChaimZhu authored
      * add pipeline
      
      * fix typos
      
      * fix unittest
      
      * fix_typos
      
      * fix_typos
      
      * change pipeline to func
      
      * refactor
      
      * refine gen indices
      
      * add device to gen indices
      
      * change tunc_objs_handle to handle_objs
      
      * fix comments
      
      * add numpy and tensor time comparision
      
      * fix
      
      * add numba accelerate
      
      * fix format
      
      * remove unnecssary func
      
      * update edge_indices docstrings
      
      * fix some comments
      c6c3c46d
  6. 01 Nov, 2021 1 commit
    • Tai-Wang's avatar
      [Feature] Support PGD Head (#964) · 36aa4cb2
      Tai-Wang 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: default avatarYezhen Cong <52420115+THU17cyz@users.noreply.github.com>
      Co-authored-by: default avatarTHU17cyz <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: default avatarYezhen Cong <52420115+THU17cyz@users.noreply.github.com>
      Co-authored-by: default avatarTHU17cyz <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
      
      * [Feature] PointXYZWHLRBBoxCoder (#856)
      
      * support PointBasedBoxCoder
      
      * fix unittest bug
      
      * support unittest in gpu
      
      * support unittest in gpu
      
      * modified docstring
      
      * add args
      
      * add args
      
      * [Enhance] Change Groupfree3D config (#855)
      
      * All mods
      
      * PointSample
      
      * PointSample
      
      * [Doc] Add tutorials/data_pipeline Chinese version (#827)
      
      * [Doc] Add tutorials/data_pipeline Chinese version
      
      * refine doc
      
      * Use the absolute link
      
      * Use the absolute link
      Co-authored-by: default avatarTai-Wang <tab_wang@outlook.com>
      
      * [Doc] Add Chinese doc for `scannet_det.md` (#836)
      
      * Part
      
      * Complete
      
      * Fix comments
      
      * Fix comments
      
      * [Doc] Add Chinese doc for `waymo_det.md` (#859)
      
      * Add complete translation
      
      * Refinements
      
      * Fix comments
      
      * Fix a minor typo
      Co-authored-by: default avatarTai-Wang <tab_wang@outlook.com>
      
      * Remove 2D annotations on Lyft (#867)
      
      * Add header for files (#869)
      
      * Add header for files
      
      * Add header for files
      
      * Add header for files
      
      * Add header for files
      
      * [fix] fix typos (#872)
      
      * Fix 3 unworking configs (#882)
      
      * [Fix] Fix `index.rst` for Chinese docs (#873)
      
      * Fix index.rst for zh docs
      
      * Change switch language
      
      * [Fix] Centerpoint head nested list transpose  (#879)
      
      * FIX Transpose nested lists without Numpy
      
      * Removed unused Numpy import
      
      * [Enhance] Update PointFusion (#791)
      
      * update point fusion
      
      * remove LIDAR hardcode
      
      * move get_proj_mat_by_coord_type to utils
      
      * fix lint
      
      * remove todo
      
      * fix lint
      
      * [Doc] Add nuscenes_det.md Chinese version (#854)
      
      * add nus chinese doc
      
      * add nuScenes Chinese doc
      
      * fix typo
      
      * fix typo
      
      * fix typo
      
      * fix typo
      
      * fix typo
      
      * [Fix] Fix RegNet pretrained weight loading (#889)
      
      * Fix regnet pretrained weight loading
      
      * Remove unused file
      
      * Fix centerpoint tta (#892)
      
      * [Enhance] Add benchmark regression script (#808)
      
      * Initial commit
      
      * [Feature] Support DGCNN (v1.0.0.dev0) (#896)
      
      * 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
      
      * Change cam rot_3d_in_axis (#906)
      
      * [Doc] Add coord sys tutorial pic and change links to dev branch (#912)
      
      * Modify link branch and add pic
      
      * Fix pic
      
      * [Feature] add kitti AP40 evaluation metric (v1.0.0.dev0) (#927)
      
      * Add citation (#901)
      
      * [Feature] Add python3.9 in CI (#900)
      
      * Add python3.0 in CI
      
      * Add python3.0 in CI
      
      * Bump to v0.17.0 (#898)
      
      * Update README.md
      
      * Update README_zh-CN.md
      
      * Update version.py
      
      * Update getting_started.md
      
      * Update getting_started.md
      
      * Update changelog.md
      
      * Remove "recent" in the news
      
      * Remove "recent" in the news
      
      * Fix comments
      
      * [Docs] Fix the version of sphinx (#902)
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * add AP40
      
      * add unitest
      
      * add unitest
      
      * seperate AP11 and AP40
      
      * fix some typos
      Co-authored-by: default avatardingchang <hudingchang.vendor@sensetime.com>
      Co-authored-by: default avatarTai-Wang <tab_wang@outlook.com>
      
      * [Feature] add smoke backbone neck (#939)
      
      * add smoke detecotor and it's backbone and neck
      
      * typo fix
      
      * fix typo
      
      * add docstring
      
      * fix typo
      
      * fix comments
      
      * fix comments
      
      * fix comments
      
      * fix typo
      
      * fix typo
      
      * fix
      
      * fix typo
      
      * fix docstring
      
      * refine feature
      
      * fix typo
      
      * use Basemodule in Neck
      
      * [Refactor] Refactor the transformation from image to camera coordinates (#938)
      
      * Refactor points_img2cam
      
      * Refine docstring
      
      * Support array converter and add unit tests
      
      * [Feature] FCOS3D BBox Coder (#940)
      
      * FCOS3D BBox Coder
      
      * Add unit tests
      
      * Change the value from long to float/double
      
      * Rename bbox_out as bbox
      
      * Add comments to forward returns
      
      * Support PGD BBox Coder
      
      * Refine docstring
      
      * Add uncertain l1 loss and its unit tests
      
      * [Feature] PGD BBox Coder (#948)
      
      * Support PGD BBox Coder
      
      * Refine docstring
      
      * PGD Head initialized
      
      * Refactor init methods, fix legacy variable names
      
      * [Feature] Support Uncertain L1 Loss (#950)
      
      * Add uncertain l1 loss and its unit tests
      
      * Remove mmcv.jit and refine docstrings
      
      * [Fix] Fix visualization in KITTI dataset (#956)
      
      * fix bug to support kitti vis
      
      * fix
      
      * Refine variable names and docstrings
      
      * Add unit tests and fix some minor bugs
      
      * Refine assertion messages
      
      * Fix typo in the docs_zh-CN
      
      * Use Pretrain init and remove unused init_cfg in FCOS3D
      
      * Fix the comments for the input_modality in the dataset config
      
      * Fix minor bugs in pgd_bbox_coder and incorrect setting for uncertain loss, use original init
      
      * Add explanations for code_weights
      
      * Adjust the unit test for pgd bbox coder
      
      * Remove unused codes
      
      * Add mono3d metric into the gather_models and fix bugs
      
      * Involve the value assignment of loss_dict into the computing procedure
      
      * Fix incorrect loss_depth
      Co-authored-by: default avatarYezhen Cong <52420115+THU17cyz@users.noreply.github.com>
      Co-authored-by: default avatarXi Liu <75658786+xiliu8006@users.noreply.github.com>
      Co-authored-by: default avatarTHU17cyz <congyezhen71@hotmail.com>
      Co-authored-by: default avatarWenhao Wu <79644370+wHao-Wu@users.noreply.github.com>
      Co-authored-by: default avatardingchang <hudingchang.vendor@sensetime.com>
      Co-authored-by: default avatar谢恩泽 <Johnny_ez@163.com>
      Co-authored-by: default avatarRobin Karlsson <34254153+robin-karlsson0@users.noreply.github.com>
      Co-authored-by: default avatarDanila Rukhovich <danrukh@gmail.com>
      Co-authored-by: default avatarChaimZhu <zhuchenming@pjlab.org.cn>
      36aa4cb2
  7. 13 Oct, 2021 1 commit
  8. 29 Sep, 2021 1 commit
    • ChaimZhu's avatar
      [Feature] Add SMOKE Head and Coder(v1.0.0.dev0) (#959) · f268ba42
      ChaimZhu 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: default avatarYezhen Cong <52420115+THU17cyz@users.noreply.github.com>
      Co-authored-by: default avatarTHU17cyz <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: default avatarYezhen Cong <52420115+THU17cyz@users.noreply.github.com>
      Co-authored-by: default avatarTHU17cyz <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
      
      * [Feature] PointXYZWHLRBBoxCoder (#856)
      
      * support PointBasedBoxCoder
      
      * fix unittest bug
      
      * support unittest in gpu
      
      * support unittest in gpu
      
      * modified docstring
      
      * add args
      
      * add args
      
      * [Enhance] Change Groupfree3D config (#855)
      
      * All mods
      
      * PointSample
      
      * PointSample
      
      * [Doc] Add tutorials/data_pipeline Chinese version (#827)
      
      * [Doc] Add tutorials/data_pipeline Chinese version
      
      * refine doc
      
      * Use the absolute link
      
      * Use the absolute link
      Co-authored-by: default avatarTai-Wang <tab_wang@outlook.com>
      
      * [Doc] Add Chinese doc for `scannet_det.md` (#836)
      
      * Part
      
      * Complete
      
      * Fix comments
      
      * Fix comments
      
      * [Doc] Add Chinese doc for `waymo_det.md` (#859)
      
      * Add complete translation
      
      * Refinements
      
      * Fix comments
      
      * Fix a minor typo
      Co-authored-by: default avatarTai-Wang <tab_wang@outlook.com>
      
      * Remove 2D annotations on Lyft (#867)
      
      * Add header for files (#869)
      
      * Add header for files
      
      * Add header for files
      
      * Add header for files
      
      * Add header for files
      
      * [fix] fix typos (#872)
      
      * Fix 3 unworking configs (#882)
      
      * [Fix] Fix `index.rst` for Chinese docs (#873)
      
      * Fix index.rst for zh docs
      
      * Change switch language
      
      * [Fix] Centerpoint head nested list transpose  (#879)
      
      * FIX Transpose nested lists without Numpy
      
      * Removed unused Numpy import
      
      * [Enhance] Update PointFusion (#791)
      
      * update point fusion
      
      * remove LIDAR hardcode
      
      * move get_proj_mat_by_coord_type to utils
      
      * fix lint
      
      * remove todo
      
      * fix lint
      
      * [Doc] Add nuscenes_det.md Chinese version (#854)
      
      * add nus chinese doc
      
      * add nuScenes Chinese doc
      
      * fix typo
      
      * fix typo
      
      * fix typo
      
      * fix typo
      
      * fix typo
      
      * [Fix] Fix RegNet pretrained weight loading (#889)
      
      * Fix regnet pretrained weight loading
      
      * Remove unused file
      
      * Fix centerpoint tta (#892)
      
      * [Enhance] Add benchmark regression script (#808)
      
      * Initial commit
      
      * [Feature] Support DGCNN (v1.0.0.dev0) (#896)
      
      * 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
      
      * Change cam rot_3d_in_axis (#906)
      
      * [Doc] Add coord sys tutorial pic and change links to dev branch (#912)
      
      * Modify link branch and add pic
      
      * Fix pic
      
      * [Feature] add kitti AP40 evaluation metric (v1.0.0.dev0) (#927)
      
      * Add citation (#901)
      
      * [Feature] Add python3.9 in CI (#900)
      
      * Add python3.0 in CI
      
      * Add python3.0 in CI
      
      * Bump to v0.17.0 (#898)
      
      * Update README.md
      
      * Update README_zh-CN.md
      
      * Update version.py
      
      * Update getting_started.md
      
      * Update getting_started.md
      
      * Update changelog.md
      
      * Remove "recent" in the news
      
      * Remove "recent" in the news
      
      * Fix comments
      
      * [Docs] Fix the version of sphinx (#902)
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * Fix sphinx version
      
      * add AP40
      
      * add unitest
      
      * add unitest
      
      * seperate AP11 and AP40
      
      * fix some typos
      Co-authored-by: default avatardingchang <hudingchang.vendor@sensetime.com>
      Co-authored-by: default avatarTai-Wang <tab_wang@outlook.com>
      
      * [Feature] add smoke backbone neck (#939)
      
      * add smoke detecotor and it's backbone and neck
      
      * typo fix
      
      * fix typo
      
      * add docstring
      
      * fix typo
      
      * fix comments
      
      * fix comments
      
      * fix comments
      
      * fix typo
      
      * fix typo
      
      * fix
      
      * fix typo
      
      * fix docstring
      
      * refine feature
      
      * fix typo
      
      * use Basemodule in Neck
      
      * [Refactor] Refactor the transformation from image to camera coordinates (#938)
      
      * Refactor points_img2cam
      
      * Refine docstring
      
      * Support array converter and add unit tests
      
      * [Feature] FCOS3D BBox Coder (#940)
      
      * FCOS3D BBox Coder
      
      * Add unit tests
      
      * Change the value from long to float/double
      
      * Rename bbox_out as bbox
      
      * Add comments to forward returns
      
      * [Feature] PGD BBox Coder (#948)
      
      * Support PGD BBox Coder
      
      * Refine docstring
      
      * [Feature] Support Uncertain L1 Loss (#950)
      
      * Add uncertain l1 loss and its unit tests
      
      * Remove mmcv.jit and refine docstrings
      
      * [Fix] Fix visualization in KITTI dataset (#956)
      
      * fix bug to support kitti vis
      
      * fix
      
      * add smoke head
      
      * fix typo
      
      * fix code
      
      * fix
      
      * fix typo
      
      * add detector
      
      * delete dectector
      
      * fix test_heads
      
      * add coder test
      
      * fix head
      
      * fix bugs in smoke head
      
      * refine kitti_mono3d data config
      
      * remove cuda is available
      
      * fix docstring
      
      * fix unitest
      
      * fix typo
      
      * refine
      
      * add affine aug
      
      * fix rebase typo
      
      * fix docs
      
      * change cam_intrinsics to cam2imgs
      
      * fix typos
      
      * fix lint
      
      * fix bugs
      
      * fix las typos
      
      * fix typos
      
      * add dosctrings for trans_mats
      Co-authored-by: default avatarYezhen Cong <52420115+THU17cyz@users.noreply.github.com>
      Co-authored-by: default avatarXi Liu <75658786+xiliu8006@users.noreply.github.com>
      Co-authored-by: default avatarTHU17cyz <congyezhen71@hotmail.com>
      Co-authored-by: default avatarWenhao Wu <79644370+wHao-Wu@users.noreply.github.com>
      Co-authored-by: default avatarTai-Wang <tab_wang@outlook.com>
      Co-authored-by: default avatardingchang <hudingchang.vendor@sensetime.com>
      Co-authored-by: default avatar谢恩泽 <Johnny_ez@163.com>
      Co-authored-by: default avatarRobin Karlsson <34254153+robin-karlsson0@users.noreply.github.com>
      Co-authored-by: default avatarDanila Rukhovich <danrukh@gmail.com>
      f268ba42
  9. 24 Sep, 2021 7 commits
  10. 22 Sep, 2021 1 commit
  11. 19 Aug, 2021 1 commit
  12. 05 Aug, 2021 1 commit
    • Wenhao Wu's avatar
      [Enhance] Sampling points based on distance metric (#667) · fc9e0d9d
      Wenhao Wu authored
      * [Enhance] Sampling points based on distance metric
      
      * fix typo
      
      * refine unittest
      
      * refine unittest
      
      * refine details & add unittest & refine configs
      
      * remove __repr__ & rename arg
      
      * fix unittest
      
      * add unitest
      
      * refine unittest
      
      * refine code
      
      * refine code
      
      * refine depth calculation
      
      * refine code
      fc9e0d9d
  13. 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
  14. 11 May, 2021 1 commit
  15. 07 Apr, 2021 1 commit
    • 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
  16. 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
  17. 17 Feb, 2021 1 commit
  18. 10 Feb, 2021 1 commit
    • xiliu8006's avatar
      [Enhance]Unittest reconstruction (#303) · 7f9b186e
      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
      7f9b186e