Update app.py

app.py

@@ -56,6 +56,7 @@ net.eval()
 
 def detect_corn(inp):
     inp = cv2.cvtColor(inp, cv2.COLOR_BGR2RGB)
+    print(inp.shape)
     frame = torch.from_numpy(inp).cpu().float()
     batch = FastBaseTransform()(frame.unsqueeze(0))
     preds = net(batch)
@@ -74,9 +75,10 @@ def detect_corn(inp):
         color = random.sample(range(0, 255), 3)
         color_mask = random.sample(range(0, 255), 3)
         cv2.rectangle(inp, (x1, y1), (x2, y2), color, int(inp.shape[0] * 0.01))
-        inp = apply_mask(inp, mask_to_save[:, :, i], [255, 0, 255], 0.5)
-    inp = cv2.resize(inp, (inp.shape[1] // 2, inp.shape[0] // 2))
+        inp = apply_mask(inp, mask_to_save[:, :, i], [0, 0, 255], 0.5)
+    inp = cv2.resize(inp, (inp.shape[1] // 4, inp.shape[0] // 4))
     inp = cv2.cvtColor(inp, cv2.COLOR_BGR2RGB)
+    print(inp.shape)
     return inp / 255.