Update run.py

run.py

@@ -40,6 +40,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, imout.shape[::-1])
     imout = im.copy()
     imout[..., 0] = imout[..., 0] + logits_n / 10
     return imout  #, cnt  #np.flipud(im)