Skip to content

Commit d82fbec

Browse files
authored
Update retinanet_presets.py (#2157)
* Update retinanet_presets.py * nit
1 parent 6ac9222 commit d82fbec

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

keras_hub/src/models/retinanet/retinanet_presets.py

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,17 @@
1111
"params": 34121239,
1212
"path": "retinanet",
1313
},
14-
"kaggle_handle": "kaggle://keras/retinanet/keras/retinanet_resnet50_fpn_coco/2",
15-
}
14+
"kaggle_handle": "kaggle://keras/retinanet/keras/retinanet_resnet50_fpn_coco/3",
15+
},
16+
"retinanet_resnet50_fpn_v2_coco": {
17+
"metadata": {
18+
"description": (
19+
"RetinaNet model with ResNet50 backbone fine-tuned on COCO in "
20+
"800x800 resolution with FPN features created from P5 level."
21+
),
22+
"params": 31558592,
23+
"path": "retinanet",
24+
},
25+
"kaggle_handle": "kaggle://keras/retinanet/keras/retinanet_resnet50_fpn_v2_coco/2",
26+
},
1627
}

0 commit comments

Comments
 (0)