cheng-hust commited on
Commit
9585432
·
verified ·
1 Parent(s): 55489ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ model2 = Model(confg='./configs/rtdetr/rtdetr_r101vd_6x_coco.yml',ckpt="./checkp
73
  #img = Image.open('./a.jpg').convert('RGB').resize((640,640))
74
 
75
 
76
- def detect(img,thr=0.2):
77
  #print(img) #ndarray
78
  img = Image.fromarray(img).resize((640,640))
79
  t_img = transformer(img).unsqueeze(0)#.unsqueeze(0) #[1,1,640,640]
 
73
  #img = Image.open('./a.jpg').convert('RGB').resize((640,640))
74
 
75
 
76
+ def detect(img,thr=0.2,cope='none'):
77
  #print(img) #ndarray
78
  img = Image.fromarray(img).resize((640,640))
79
  t_img = transformer(img).unsqueeze(0)#.unsqueeze(0) #[1,1,640,640]