wenpeng commited on
Commit
6483e22
·
1 Parent(s): 7079dcb

add sod output

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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:,:]/255))
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