Skip to content

Commit b0e2687

Browse files
ppwwyyxxfacebook-github-bot
authored andcommitted
use L1 loss in RetinaNet
Summary: update model zoo to be consistent with R-CNN Reviewed By: rbgirshick Differential Revision: D21614321 fbshipit-source-id: 4b1147be725f1f9335d51b5404d12d4d3093adb2
1 parent 5d91693 commit b0e2687

File tree

4 files changed

+25
-24
lines changed

4 files changed

+25
-24
lines changed

MODEL_ZOO.md

Lines changed: 19 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,6 @@ All models available for download through this document are licensed under the
202202
./gen_html_table.py --config 'COCO-Detection/retina*50*' 'COCO-Detection/retina*101*' --name R50 R50 R101 --fields lr_sched train_speed inference_speed mem box_AP
203203
-->
204204

205-
206205
<table><tbody>
207206
<!-- START TABLE -->
208207
<!-- TABLE HEADER -->
@@ -218,35 +217,36 @@ All models available for download through this document are licensed under the
218217
<!-- ROW: retinanet_R_50_FPN_1x -->
219218
<tr><td align="left"><a href="configs/COCO-Detection/retinanet_R_50_FPN_1x.yaml">R50</a></td>
220219
<td align="center">1x</td>
221-
<td align="center">0.200</td>
222-
<td align="center">0.055</td>
223-
<td align="center">3.9</td>
224-
<td align="center">36.5</td>
225-
<td align="center">137593951</td>
226-
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_50_FPN_1x/137593951/model_final_b796dc.pkl">model</a>&nbsp;|&nbsp;<a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_50_FPN_1x/137593951/metrics.json">metrics</a></td>
220+
<td align="center">0.205</td>
221+
<td align="center">0.056</td>
222+
<td align="center">4.1</td>
223+
<td align="center">37.4</td>
224+
<td align="center">190397773</td>
225+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_50_FPN_1x/190397773/model_final_bfca0b.pkl">model</a>&nbsp;|&nbsp;<a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_50_FPN_1x/190397773/metrics.json">metrics</a></td>
227226
</tr>
228227
<!-- ROW: retinanet_R_50_FPN_3x -->
229228
<tr><td align="left"><a href="configs/COCO-Detection/retinanet_R_50_FPN_3x.yaml">R50</a></td>
230229
<td align="center">3x</td>
231-
<td align="center">0.201</td>
232-
<td align="center">0.055</td>
233-
<td align="center">3.9</td>
234-
<td align="center">37.9</td>
235-
<td align="center">137849486</td>
236-
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_50_FPN_3x/137849486/model_final_4cafe0.pkl">model</a>&nbsp;|&nbsp;<a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_50_FPN_3x/137849486/metrics.json">metrics</a></td>
230+
<td align="center">0.205</td>
231+
<td align="center">0.056</td>
232+
<td align="center">4.1</td>
233+
<td align="center">38.7</td>
234+
<td align="center">190397829</td>
235+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_50_FPN_3x/190397829/model_final_5bd44e.pkl">model</a>&nbsp;|&nbsp;<a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_50_FPN_3x/190397829/metrics.json">metrics</a></td>
237236
</tr>
238237
<!-- ROW: retinanet_R_101_FPN_3x -->
239238
<tr><td align="left"><a href="configs/COCO-Detection/retinanet_R_101_FPN_3x.yaml">R101</a></td>
240239
<td align="center">3x</td>
241-
<td align="center">0.280</td>
242-
<td align="center">0.068</td>
243-
<td align="center">5.1</td>
244-
<td align="center">39.9</td>
245-
<td align="center">138363263</td>
246-
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_101_FPN_3x/138363263/model_final_59f53c.pkl">model</a>&nbsp;|&nbsp;<a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_101_FPN_3x/138363263/metrics.json">metrics</a></td>
240+
<td align="center">0.291</td>
241+
<td align="center">0.069</td>
242+
<td align="center">5.2</td>
243+
<td align="center">40.4</td>
244+
<td align="center">190397697</td>
245+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_101_FPN_3x/190397697/model_final_971ab9.pkl">model</a>&nbsp;|&nbsp;<a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_101_FPN_3x/190397697/metrics.json">metrics</a></td>
247246
</tr>
248247
</tbody></table>
249248

249+
250250
#### RPN & Fast R-CNN:
251251
<!--
252252
./gen_html_table.py --config 'COCO-Detection/rpn*' 'COCO-Detection/fast_rcnn*' --name "RPN R50-C4" "RPN R50-FPN" "Fast R-CNN R50-FPN" --fields lr_sched train_speed inference_speed mem box_AP prop_AR

configs/Base-RetinaNet.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ MODEL:
1111
RETINANET:
1212
IOU_THRESHOLDS: [0.4, 0.5]
1313
IOU_LABELS: [0, -1, 1]
14+
SMOOTH_L1_LOSS_BETA: 0.0
1415
DATASETS:
1516
TRAIN: ("coco_2017_train",)
1617
TEST: ("coco_2017_val",)
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
_BASE_: "../COCO-Detection/retinanet_R_50_FPN_3x.yaml"
22
MODEL:
3-
WEIGHTS: "detectron2://COCO-Detection/retinanet_R_50_FPN_3x/137849486/model_final_4cafe0.pkl"
3+
WEIGHTS: "detectron2://COCO-Detection/retinanet_R_50_FPN_3x/190397829/model_final_5bd44e.pkl"
44
DATASETS:
55
TEST: ("coco_2017_val_100",)
66
TEST:
7-
EXPECTED_RESULTS: [["bbox", "AP", 44.36, 0.02]]
7+
EXPECTED_RESULTS: [["bbox", "AP", 44.45, 0.02]]

detectron2/model_zoo/model_zoo.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,9 @@ class _ModelZooUrls(object):
2929
"COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml": "137851257/model_final_f6e8b1.pkl",
3030
"COCO-Detection/faster_rcnn_X_101_32x8d_FPN_3x.yaml": "139173657/model_final_68b088.pkl",
3131
# COCO Detection with RetinaNet
32-
"COCO-Detection/retinanet_R_50_FPN_1x.yaml": "137593951/model_final_b796dc.pkl",
33-
"COCO-Detection/retinanet_R_50_FPN_3x.yaml": "137849486/model_final_4cafe0.pkl",
34-
"COCO-Detection/retinanet_R_101_FPN_3x.yaml": "138363263/model_final_59f53c.pkl",
32+
"COCO-Detection/retinanet_R_50_FPN_1x.yaml": "190397773/model_final_bfca0b.pkl",
33+
"COCO-Detection/retinanet_R_50_FPN_3x.yaml": "190397829/model_final_5bd44e.pkl",
34+
"COCO-Detection/retinanet_R_101_FPN_3x.yaml": "190397697/model_final_971ab9.pkl",
3535
# COCO Detection with RPN and Fast R-CNN
3636
"COCO-Detection/rpn_R_50_C4_1x.yaml": "137258005/model_final_450694.pkl",
3737
"COCO-Detection/rpn_R_50_FPN_1x.yaml": "137258492/model_final_02ce48.pkl",

0 commit comments

Comments
 (0)