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