Alesteba commited on
Commit
e206b35
·
1 Parent(s): 0ca578a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.resnet50_fpn (pretrained=True),
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