Spaces:

oliverlevn
/

ocean_faster_RCNN

Build error

oliverlevn commited on Jul 4, 2023

Commit

bb029d9

1 Parent(s): 3a42afa

remove resize

Files changed (1) hide show

app.py CHANGED Viewed

@@ -77,7 +77,7 @@ def predict(img):
     # Start the timer
     start_time = time.time()
     device = 'cpu'
-    transform = T.Compose([T.ToPILImage(), T.Resize(size = (768,1024)),T.ToTensor()])
     image_tensor = transform(img).to(device)
     image_tensor = image_tensor.unsqueeze(0)
     faster_rcnn.eval()

     # Start the timer
     start_time = time.time()
     device = 'cpu'
+    transform = T.Compose([T.ToPILImage(),T.ToTensor()])
     image_tensor = transform(img).to(device)
     image_tensor = image_tensor.unsqueeze(0)
     faster_rcnn.eval()