metafile.yml 2.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. Collections:
  2. - Name: CornerNet
  3. Metadata:
  4. Training Data: COCO
  5. Training Techniques:
  6. - Adam
  7. Training Resources: 8x V100 GPUs
  8. Architecture:
  9. - Corner Pooling
  10. - Stacked Hourglass Network
  11. Paper:
  12. URL: https://arxiv.org/abs/1808.01244
  13. Title: 'CornerNet: Detecting Objects as Paired Keypoints'
  14. README: configs/cornernet/README.md
  15. Code:
  16. URL: https://github.com/open-mmlab/mmdetection/blob/v2.3.0/mmdet/models/detectors/cornernet.py#L9
  17. Version: v2.3.0
  18. Models:
  19. - Name: cornernet_hourglass104_10xb5-crop511-210e-mstest_coco
  20. In Collection: CornerNet
  21. Config: configs/cornernet/cornernet_hourglass104_10xb5-crop511-210e-mstest_coco.py
  22. Metadata:
  23. Training Resources: 10x V100 GPUs
  24. Batch Size: 50
  25. Training Memory (GB): 13.9
  26. inference time (ms/im):
  27. - value: 238.1
  28. hardware: V100
  29. backend: PyTorch
  30. batch size: 1
  31. mode: FP32
  32. resolution: (800, 1333)
  33. Epochs: 210
  34. Results:
  35. - Task: Object Detection
  36. Dataset: COCO
  37. Metrics:
  38. box AP: 41.2
  39. Weights: https://download.openmmlab.com/mmdetection/v2.0/cornernet/cornernet_hourglass104_mstest_10x5_210e_coco/cornernet_hourglass104_mstest_10x5_210e_coco_20200824_185720-5fefbf1c.pth
  40. - Name: cornernet_hourglass104_8xb6-210e-mstest_coco
  41. In Collection: CornerNet
  42. Config: configs/cornernet/cornernet_hourglass104_8xb6-210e-mstest_coco.py
  43. Metadata:
  44. Batch Size: 48
  45. Training Memory (GB): 15.9
  46. inference time (ms/im):
  47. - value: 238.1
  48. hardware: V100
  49. backend: PyTorch
  50. batch size: 1
  51. mode: FP32
  52. resolution: (800, 1333)
  53. Epochs: 210
  54. Results:
  55. - Task: Object Detection
  56. Dataset: COCO
  57. Metrics:
  58. box AP: 41.2
  59. Weights: https://download.openmmlab.com/mmdetection/v2.0/cornernet/cornernet_hourglass104_mstest_8x6_210e_coco/cornernet_hourglass104_mstest_8x6_210e_coco_20200825_150618-79b44c30.pth
  60. - Name: cornernet_hourglass104_32xb3-210e-mstest_coco
  61. In Collection: CornerNet
  62. Config: configs/cornernet/cornernet_hourglass104_32xb3-210e-mstest_coco.py
  63. Metadata:
  64. Training Resources: 32x V100 GPUs
  65. Batch Size: 96
  66. Training Memory (GB): 9.5
  67. inference time (ms/im):
  68. - value: 256.41
  69. hardware: V100
  70. backend: PyTorch
  71. batch size: 1
  72. mode: FP32
  73. resolution: (800, 1333)
  74. Epochs: 210
  75. Results:
  76. - Task: Object Detection
  77. Dataset: COCO
  78. Metrics:
  79. box AP: 40.4
  80. Weights: https://download.openmmlab.com/mmdetection/v2.0/cornernet/cornernet_hourglass104_mstest_32x3_210e_coco/cornernet_hourglass104_mstest_32x3_210e_coco_20200819_203110-1efaea91.pth