Skip to content

Commit dbfc652

Browse files
Added EfficientDet AdvProp-AA D0-D5 (airctic#870)
* added EfficientDet AdvProp-AA D0-D5 * added test for EfficientDet AdvProp-AA models * prefix with tf ap models
1 parent 416555e commit dbfc652

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed

icevision/models/ross/efficientdet/backbones.py

+13
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,12 @@
2121
"d6",
2222
"d7",
2323
"d7x",
24+
"tf_d0_ap",
25+
"tf_d1_ap",
26+
"tf_d2_ap",
27+
"tf_d3_ap",
28+
"tf_d4_ap",
29+
"tf_d5_ap",
2430
]
2531

2632
from icevision.models.ross.efficientdet.utils import *
@@ -50,3 +56,10 @@
5056
d6 = EfficientDetBackboneConfig(model_name="efficientdet_d6")
5157
d7 = EfficientDetBackboneConfig(model_name="efficientdet_d7")
5258
d7x = EfficientDetBackboneConfig(model_name="efficientdet_d7x")
59+
60+
tf_d0_ap = EfficientDetBackboneConfig(model_name="tf_efficientdet_d0_ap")
61+
tf_d1_ap = EfficientDetBackboneConfig(model_name="tf_efficientdet_d1_ap")
62+
tf_d2_ap = EfficientDetBackboneConfig(model_name="tf_efficientdet_d2_ap")
63+
tf_d3_ap = EfficientDetBackboneConfig(model_name="tf_efficientdet_d3_ap")
64+
tf_d4_ap = EfficientDetBackboneConfig(model_name="tf_efficientdet_d4_ap")
65+
tf_d5_ap = EfficientDetBackboneConfig(model_name="tf_efficientdet_d5_ap")

tests/models/efficient_det/test_model.py

+3
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,9 @@
1010
d0,
1111
d1,
1212
d2,
13+
tf_d0_ap,
14+
tf_d1_ap,
15+
tf_d2_ap,
1316
tf_d0,
1417
tf_d1,
1518
tf_d2,

0 commit comments

Comments
 (0)