cheng-hust commited on
Commit
c09a5d2
·
verified ·
1 Parent(s): 28b4a8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 cope == 'cope12':
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)