• hjin2902's avatar
    [Feature] Group-Free-3D head (#539) · 318499ac
    hjin2902 authored
    
    
    * group-free-3d head
    
    * GroupFree3DNet->VoteNet
    
    * modify docstring
    
    * bugfix: calculate pts_instance_label, decoder self/cross posembed init
    
    * support GroupFree3DNet, modify scannnet train config
    
    * support point cloud input features dim = 0
    
    * add groupfree3dnet test case
    
    * bugfix: softmax in decode boxes
    
    * support multi-stage predictions
    
    * modify GroupFree3DMultiheadAttention input parameters
    
    * refactor: support sunrgbd-based train
    
    * refactor: support sunrgbd-based train
    
    * fix parts of bug
    
    * modify multi-stage prediction
    
    * fixbug: conv_channels
    
    * bugfix: permute
    
    * bugfix: permute
    
    * bugfix: expand
    
    * fix MAX_NUM_OBJ=64
    
    * 4 gpu training, score_thr = 0.0
    
    * modify config, repeattime=1
    
    * bigfix: expand
    
    * modify: GroupFree3DMHA, build_positional_encoding
    
    * modify: GroupFree3DMHA, build_positional_encoding
    
    * bugfix: torch.nn
    
    * bugfix: mean loss
    
    * residual -> identity
    
    * fix name: DropOut -> Dropout
    
    * delete sunrgbd-based congfig
    
    * Fix: trailing whitespace
    
    * suffix -> prefix
    
    * bugfix: groupfree3d config
    Co-authored-by: default avatarjinhui <PJLAB\jinhui@shai14001019l.pjlab.org>
    318499ac
test_detectors.py 17.5 KB