Update app.py
Browse files
app.py
CHANGED
@@ -81,7 +81,7 @@ def detect(img,thr=0.2,trained_dataset='aitod'):
|
|
81 |
#print(t_img.shape)
|
82 |
if trained_dataset == 'aitod':
|
83 |
labels, boxes, scores=model(t_img,size)
|
84 |
-
elif
|
85 |
labels, boxes, scores=model2(t_img,size)
|
86 |
else:
|
87 |
labels, boxes, scores=model3(t_img,size)
|
|
|
81 |
#print(t_img.shape)
|
82 |
if trained_dataset == 'aitod':
|
83 |
labels, boxes, scores=model(t_img,size)
|
84 |
+
elif trained_dataset == 'cope12':
|
85 |
labels, boxes, scores=model2(t_img,size)
|
86 |
else:
|
87 |
labels, boxes, scores=model3(t_img,size)
|