Update run.py
Browse files
run.py
CHANGED
@@ -41,7 +41,7 @@ def segment(im, interval_s=2):
|
|
41 |
logits = outputs.logits.cpu().detach().numpy() ** 2
|
42 |
logits_n = (logits[0, 0] - logits[0, 0].min()) / (logits[0, 0].max() - logits[0, 0].min())
|
43 |
imout = im.copy()
|
44 |
-
imout[..., 0] =
|
45 |
return imout #, cnt #np.flipud(im)
|
46 |
|
47 |
with gr.Blocks() as demo:
|
|
|
41 |
logits = outputs.logits.cpu().detach().numpy() ** 2
|
42 |
logits_n = (logits[0, 0] - logits[0, 0].min()) / (logits[0, 0].max() - logits[0, 0].min())
|
43 |
imout = im.copy()
|
44 |
+
imout[..., 0] = imout[..., 0] + logits_n / 10
|
45 |
return imout #, cnt #np.flipud(im)
|
46 |
|
47 |
with gr.Blocks() as demo:
|