Update app.py
Browse files
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]
|