Update app.py

app.py

@@ -11,7 +11,7 @@ from pathlib import Path
 
 transformer = transforms.Compose([    
     transforms.ToTensor(),                            
-    transforms.Resize([640,640]),
+    #transforms.Resize([640,640]),
 
 ])
 
@@ -68,10 +68,9 @@ model = Model(confg='./configs/rtdetr/rtdetr_r101vd_6x_coco.yml',ckpt="./checkpo
 #img = Image.open('./a.jpg').convert('RGB').resize((640,640))
 
 
-
 def detect(img):
-  print(img)
-  #img = img.resize((640,640))
+  #print(img)                    #ndarray
+  img = Image.fromarray(img).resize((640,640))
   t_img = transformer(img).unsqueeze(0)#.unsqueeze(0)      #[1,1,640,640]
   size = torch.tensor([[t_img.shape[2], t_img.shape[3]]])
   #print(t_img.shape)