Update run.py

run.py

@@ -41,7 +41,7 @@ def segment(im, interval_s=2):
     outputs = model(pixel_values=pixel_values)
     logits = outputs.logits.cpu().detach().numpy() ** 2
     logits_n = (logits[0, 0] - logits[0, 0].min()) / (logits[0, 0].max() - logits[0, 0].min())
-    logits_n = cv2.resize(logits_n, (im.shape[1], im.shape[0])])
+    logits_n = cv2.resize(logits_n, (im.shape[1], im.shape[0]))
     imout = im.copy()
     imout[..., 0] = imout[..., 0] + logits_n / 10
     return imout  #, cnt  #np.flipud(im)