Spaces:
Runtime error
Runtime error
add sod output
Browse files
app.py
CHANGED
|
@@ -31,7 +31,7 @@ def sod_inpaint(img):
|
|
| 31 |
sod_res = sod_model.forward(img,None)
|
| 32 |
sod_res = np.uint8(sod_res)
|
| 33 |
h,w = sod_res.shape[:2]
|
| 34 |
-
so = np.uint8(sod_res[:,:w//2,:] * (sod_res[:,w//2:,:]
|
| 35 |
inpaint_res = inpaint_model.forward(sod_res,None)
|
| 36 |
inpaint_res = np.uint8(inpaint_res)
|
| 37 |
count +=1
|
|
|
|
| 31 |
sod_res = sod_model.forward(img,None)
|
| 32 |
sod_res = np.uint8(sod_res)
|
| 33 |
h,w = sod_res.shape[:2]
|
| 34 |
+
so = np.uint8(sod_res[:,:w//2,:] * (sod_res[:,w//2:,:]>0).astype(np.float32))
|
| 35 |
inpaint_res = inpaint_model.forward(sod_res,None)
|
| 36 |
inpaint_res = np.uint8(inpaint_res)
|
| 37 |
count +=1
|