cheng-hust commited on
Commit
4fb77db
·
verified ·
1 Parent(s): 6201c92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -68,7 +68,7 @@ model = Model(confg='./configs/rtdetr/rtdetr_r101vd_6x_coco.yml',ckpt="./checkpo
68
  #img = Image.open('./a.jpg').convert('RGB').resize((640,640))
69
 
70
 
71
- def detect(img,thr):
72
  #print(img) #ndarray
73
  img = Image.fromarray(img).resize((640,640))
74
  t_img = transformer(img).unsqueeze(0)#.unsqueeze(0) #[1,1,640,640]
 
68
  #img = Image.open('./a.jpg').convert('RGB').resize((640,640))
69
 
70
 
71
+ def detect(img,thr=0.2):
72
  #print(img) #ndarray
73
  img = Image.fromarray(img).resize((640,640))
74
  t_img = transformer(img).unsqueeze(0)#.unsqueeze(0) #[1,1,640,640]