"vscode:/vscode.git/clone" did not exist on "657863bb35bdb44ec709a6487c3b0c8cf10c976f"
  • twang's avatar
    [Feature] Add shape-aware grouping head in SSN (#147) · b016d90c
    twang authored
    * Add shape-aware grouping head
    
    * Reformat docstrings
    
    * Remove rewritten get_anchors in shape_aware_head
    
    * Refactor and simplify shape-aware grouping head
    
    * Fix docstring
    
    * Remove fixed preset shape heads
    
    * Add unittest for AlignedAnchor3DRangeGeneratorPerCls
    
    * Add unittest for get bboxes in shape_aware_head
    
    * Add unittest for loss of shape_aware_head
    
    * Fix unstandard docstrings
    
    * Minor fix for a comment
    
    * Add assertion to make sure not all boxes are filtered
    b016d90c
train_mixins.py 16.3 KB