Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ class_map = ClassMap(['raccoon','banana'])
|
|
12 |
size = 384
|
13 |
|
14 |
model_2 = models.torchvision.retinanet.model(
|
15 |
-
backbone=models.torchvision.retinanet.backbones.
|
16 |
num_classes=len(class_map)
|
17 |
)
|
18 |
|
|
|
12 |
size = 384
|
13 |
|
14 |
model_2 = models.torchvision.retinanet.model(
|
15 |
+
backbone=models.torchvision.retinanet.backbones.wide_resnet101_2_fpn (pretrained=True),
|
16 |
num_classes=len(class_map)
|
17 |
)
|
18 |
|