• 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
__init__.py 67 Bytes