• Hu Ye's avatar
    add FCOS (#4961) · 7d4bdd43
    Hu Ye authored
    
    
    * add fcos
    
    * update fcos
    
    * add giou_loss
    
    * add BoxLinearCoder for FCOS
    
    * add full code for FCOS
    
    * add giou loss
    
    * add fcos
    
    * add __all__
    
    * Fixing lint
    
    * Fixing lint in giou_loss.py
    
    * Add typing annotation to fcos
    
    * Add trained checkpoints
    
    * Use partial to replace lambda
    
    * Minor fixes to docstrings
    
    * Apply ufmt format
    
    * Fixing docstrings
    
    * Fixing jit scripting
    
    * Minor fixes to docstrings
    
    * Fixing jit scripting
    
    * Ignore mypy in fcos
    
    * Fixing trained checkpoints
    
    * Fixing unit-test of jit script
    
    * Fixing docstrings
    
    * Add test/expect/ModelTester.test_fcos_resnet50_fpn_expect.pkl
    
    * Fixing test_detection_model_trainable_backbone_layers
    
    * Update test_fcos_resnet50_fpn_expect.pkl
    
    * rename stride to box size
    
    * remove TODO and fix some typo
    
    * merge some code for better
    
    * impove the comments
    
    * remove decode and encode of BoxLinearCoder
    
    * remove some unnecessary hints
    
    * use the default value in detectron2.
    
    * update doc
    
    * Add unittest for BoxLinearCoder
    
    * Add types in FCOS
    
    * Add docstring for BoxLinearCoder
    
    * Minor fix for the docstring
    
    * update doc
    
    * Update fcos_resnet50_fpn_coco pretained weights url
    
    * Update torchvision/models/detection/fcos.py
    Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
    
    * Update torchvision/models/detection/fcos.py
    Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
    
    * Update torchvision/models/detection/fcos.py
    Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
    
    * Update torchvision/models/detection/fcos.py
    Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
    
    * Add FCOS model documentation
    
    * Fix typo in FCOS documentation
    
    * Add fcos to the prototype builder
    
    * Capitalize COCO_V1
    
    * Fix params of fcos
    
    * fix bug for partial
    
    * Fixing docs indentation
    
    * Fixing docs format in giou_loss
    
    * Adopt Reference for GIoU Loss
    
    * Rename giou_loss to generalized_box_iou_loss
    
    * remove overwrite_eps
    
    * Update AP test values
    
    * Minor fixes for the docs
    
    * Minor fixes for the docs
    
    * Update torchvision/models/detection/fcos.py
    Co-authored-by: default avatarZhiqiang Wang <zhiqwang@foxmail.com>
    
    * Update torchvision/prototype/models/detection/fcos.py
    Co-authored-by: default avatarZhiqiang Wang <zhiqwang@foxmail.com>
    Co-authored-by: default avatarzhiqiang <zhiqwang@foxmail.com>
    Co-authored-by: default avatarJoao Gomes <jdsgomes@fb.com>
    Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
    Co-authored-by: default avatarJoao Gomes <joaopsgomes@gmail.com>
    7d4bdd43
test_models_detection_utils.py 4.06 KB