metafile.yml 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. Collections:
  2. - Name: ATSS
  3. Metadata:
  4. Training Data: COCO
  5. Training Techniques:
  6. - SGD with Momentum
  7. - Weight Decay
  8. Training Resources: 8x V100 GPUs
  9. Architecture:
  10. - ATSS
  11. - FPN
  12. - ResNet
  13. Paper:
  14. URL: https://arxiv.org/abs/1912.02424
  15. Title: 'Bridging the Gap Between Anchor-based and Anchor-free Detection via Adaptive Training Sample Selection'
  16. README: configs/atss/README.md
  17. Code:
  18. URL: https://github.com/open-mmlab/mmdetection/blob/v2.0.0/mmdet/models/detectors/atss.py#L6
  19. Version: v2.0.0
  20. Models:
  21. - Name: atss_r50_fpn_1x_coco
  22. In Collection: ATSS
  23. Config: configs/atss/atss_r50_fpn_1x_coco.py
  24. Metadata:
  25. Training Memory (GB): 3.7
  26. inference time (ms/im):
  27. - value: 50.76
  28. hardware: V100
  29. backend: PyTorch
  30. batch size: 1
  31. mode: FP32
  32. resolution: (800, 1333)
  33. Epochs: 12
  34. Results:
  35. - Task: Object Detection
  36. Dataset: COCO
  37. Metrics:
  38. box AP: 39.4
  39. Weights: https://download.openmmlab.com/mmdetection/v2.0/atss/atss_r50_fpn_1x_coco/atss_r50_fpn_1x_coco_20200209-985f7bd0.pth
  40. - Name: atss_r101_fpn_1x_coco
  41. In Collection: ATSS
  42. Config: configs/atss/atss_r101_fpn_1x_coco.py
  43. Metadata:
  44. Training Memory (GB): 5.6
  45. inference time (ms/im):
  46. - value: 81.3
  47. hardware: V100
  48. backend: PyTorch
  49. batch size: 1
  50. mode: FP32
  51. resolution: (800, 1333)
  52. Epochs: 12
  53. Results:
  54. - Task: Object Detection
  55. Dataset: COCO
  56. Metrics:
  57. box AP: 41.5
  58. Weights: https://download.openmmlab.com/mmdetection/v2.0/atss/atss_r101_fpn_1x_coco/atss_r101_fpn_1x_20200825-dfcadd6f.pth