1. 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
  2. 19 Sep, 2020 1 commit
    • yinchimaoliang's avatar
      [feature]: Add multi-group head of CenterPoint (#49) · d07bd8bd
      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
      d07bd8bd
  3. 26 Apr, 2020 1 commit
  4. 14 Apr, 2020 1 commit