Spaces:

noamholz
/

gra

Sleeping

noamholz commited on Jan 21

Commit

c47ffe0

verified ·

1 Parent(s): 3476cae

Update run.py

Files changed (1) hide show

run.py CHANGED Viewed

@@ -43,7 +43,7 @@ def segment(im, interval_s=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]))
     imout = im.copy()
-    imout[..., 0] = imout[..., 0] + logits_n / 10
     return imout  #, cnt  #np.flipud(im)
 with gr.Blocks() as demo:

     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]))
     imout = im.copy()
+    imout[..., 1] = imout[..., 1] + logits_n * 200
     return imout  #, cnt  #np.flipud(im)
 with gr.Blocks() as demo: