1. 28 Mar, 2022 2 commits
  2. 01 Feb, 2022 1 commit
  3. 24 Jan, 2022 1 commit
    • Joao Gomes's avatar
      Fix giou naming bug (#5270) · abdae5a1
      Joao Gomes authored
      * Adding multiweight support for shufflenetv2 prototype models
      
      * Revert "Adding multiweight support for shufflenetv2 prototype models"
      
      This reverts commit 31fadbee7d1a65cd73ae43dfd4ac6e97e7ca7b01.
      
      * Adding multiweight support for shufflenetv2 prototype models
      
      * Revert "Adding multiweight support for shufflenetv2 prototype models"
      
      This reverts commit 4e3d900f796c1e3e667312087e77956ca4a4c017.
      
      * Remove module vs method name clash
      abdae5a1
  4. 21 Jan, 2022 1 commit
    • 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