Hugging Face's logo

Spaces:
vahidrezanezhad
/
sbb_binarization_demo
Runtime error

App Files Community
vahidrezanezhad commited on
Commit
1e70ba8
·
verified ·
1 Parent(s): 97d54df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -0
app.py CHANGED
@@ -62,24 +62,52 @@ def do_prediction(img):
62
  label_p_pred = model.predict(img_patch.reshape(1, img_patch.shape[0], img_patch.shape[1], img_patch.shape[2]),
63
  verbose=0)
64
  seg = np.argmax(label_p_pred, axis=2)
 
65
 
66
  if i == 0 and j == 0:
 
 
 
67
  prediction_true[index_y_d + 0 : index_y_u - margin, index_x_d + 0 : index_x_u - margin, :] = seg_color
68
  elif i == nxf - 1 and j == nyf - 1:
 
 
 
69
  prediction_true[index_y_d + margin : index_y_u - 0, index_x_d + margin : index_x_u - 0, :] = seg_color
70
  elif i == 0 and j == nyf - 1:
 
 
 
71
  prediction_true[index_y_d + margin : index_y_u - 0, index_x_d + 0 : index_x_u - margin, :] = seg_color
72
  elif i == nxf - 1 and j == 0:
 
 
 
73
  prediction_true[index_y_d + 0 : index_y_u - margin, index_x_d + margin : index_x_u - 0, :] = seg_color
74
  elif i == 0 and j != 0 and j != nyf - 1:
 
 
 
75
  prediction_true[index_y_d + margin : index_y_u - margin, index_x_d + 0 : index_x_u - margin, :] = seg_color
76
  elif i == nxf - 1 and j != 0 and j != nyf - 1:
 
 
 
77
  prediction_true[index_y_d + margin : index_y_u - margin, index_x_d + margin : index_x_u - 0, :] = seg_color
78
  elif i != 0 and i != nxf - 1 and j == 0:
 
 
 
79
  prediction_true[index_y_d + 0 : index_y_u - margin, index_x_d + margin : index_x_u - margin, :] = seg_color
80
  elif i != 0 and i != nxf - 1 and j == nyf - 1:
 
 
 
81
  prediction_true[index_y_d + margin : index_y_u - 0, index_x_d + margin : index_x_u - margin, :] = seg_color
82
  else:
 
 
 
83
  prediction_true[index_y_d + margin : index_y_u - margin, index_x_d + margin : index_x_u - margin, :] = seg_color
84
 
85
  prediction_true = prediction_true.astype(np.uint8)
 
62
  label_p_pred = model.predict(img_patch.reshape(1, img_patch.shape[0], img_patch.shape[1], img_patch.shape[2]),
63
  verbose=0)
64
  seg = np.argmax(label_p_pred, axis=2)
65
+ seg_color = np.repeat(seg[:, :, np.newaxis], 3, axis=2)
66
 
67
  if i == 0 and j == 0:
68
+ seg_color = seg_color[0 : seg_color.shape[0] - margin, 0 : seg_color.shape[1] - margin, :]
69
+ #seg = seg[0 : seg.shape[0] - margin, 0 : seg.shape[1] - margin]
70
+ #mask_true[index_y_d + 0 : index_y_u - margin, index_x_d + 0 : index_x_u - margin] = seg
71
  prediction_true[index_y_d + 0 : index_y_u - margin, index_x_d + 0 : index_x_u - margin, :] = seg_color
72
  elif i == nxf - 1 and j == nyf - 1:
73
+ seg_color = seg_color[margin : seg_color.shape[0] - 0, margin : seg_color.shape[1] - 0, :]
74
+ #seg = seg[margin : seg.shape[0] - 0, margin : seg.shape[1] - 0]
75
+ #mask_true[index_y_d + margin : index_y_u - 0, index_x_d + margin : index_x_u - 0] = seg
76
  prediction_true[index_y_d + margin : index_y_u - 0, index_x_d + margin : index_x_u - 0, :] = seg_color
77
  elif i == 0 and j == nyf - 1:
78
+ seg_color = seg_color[margin : seg_color.shape[0] - 0, 0 : seg_color.shape[1] - margin, :]
79
+ #seg = seg[margin : seg.shape[0] - 0, 0 : seg.shape[1] - margin]
80
+ #mask_true[index_y_d + margin : index_y_u - 0, index_x_d + 0 : index_x_u - margin] = seg
81
  prediction_true[index_y_d + margin : index_y_u - 0, index_x_d + 0 : index_x_u - margin, :] = seg_color
82
  elif i == nxf - 1 and j == 0:
83
+ seg_color = seg_color[0 : seg_color.shape[0] - margin, margin : seg_color.shape[1] - 0, :]
84
+ #seg = seg[0 : seg.shape[0] - margin, margin : seg.shape[1] - 0]
85
+ #mask_true[index_y_d + 0 : index_y_u - margin, index_x_d + margin : index_x_u - 0] = seg
86
  prediction_true[index_y_d + 0 : index_y_u - margin, index_x_d + margin : index_x_u - 0, :] = seg_color
87
  elif i == 0 and j != 0 and j != nyf - 1:
88
+ seg_color = seg_color[margin : seg_color.shape[0] - margin, 0 : seg_color.shape[1] - margin, :]
89
+ #seg = seg[margin : seg.shape[0] - margin, 0 : seg.shape[1] - margin]
90
+ #mask_true[index_y_d + margin : index_y_u - margin, index_x_d + 0 : index_x_u - margin] = seg
91
  prediction_true[index_y_d + margin : index_y_u - margin, index_x_d + 0 : index_x_u - margin, :] = seg_color
92
  elif i == nxf - 1 and j != 0 and j != nyf - 1:
93
+ seg_color = seg_color[margin : seg_color.shape[0] - margin, margin : seg_color.shape[1] - 0, :]
94
+ #seg = seg[margin : seg.shape[0] - margin, margin : seg.shape[1] - 0]
95
+ #mask_true[index_y_d + margin : index_y_u - margin, index_x_d + margin : index_x_u - 0] = seg
96
  prediction_true[index_y_d + margin : index_y_u - margin, index_x_d + margin : index_x_u - 0, :] = seg_color
97
  elif i != 0 and i != nxf - 1 and j == 0:
98
+ seg_color = seg_color[0 : seg_color.shape[0] - margin, margin : seg_color.shape[1] - margin, :]
99
+ #seg = seg[0 : seg.shape[0] - margin, margin : seg.shape[1] - margin]
100
+ #mask_true[index_y_d + 0 : index_y_u - margin, index_x_d + margin : index_x_u - margin] = seg
101
  prediction_true[index_y_d + 0 : index_y_u - margin, index_x_d + margin : index_x_u - margin, :] = seg_color
102
  elif i != 0 and i != nxf - 1 and j == nyf - 1:
103
+ seg_color = seg_color[margin : seg_color.shape[0] - 0, margin : seg_color.shape[1] - margin, :]
104
+ #seg = seg[margin : seg.shape[0] - 0, margin : seg.shape[1] - margin]
105
+ #mask_true[index_y_d + margin : index_y_u - 0, index_x_d + margin : index_x_u - margin] = seg
106
  prediction_true[index_y_d + margin : index_y_u - 0, index_x_d + margin : index_x_u - margin, :] = seg_color
107
  else:
108
+ seg_color = seg_color[margin : seg_color.shape[0] - margin, margin : seg_color.shape[1] - margin, :]
109
+ #seg = seg[margin : seg.shape[0] - margin, margin : seg.shape[1] - margin]
110
+ #mask_true[index_y_d + margin : index_y_u - margin, index_x_d + margin : index_x_u - margin] = seg
111
  prediction_true[index_y_d + margin : index_y_u - margin, index_x_d + margin : index_x_u - margin, :] = seg_color
112
 
113
  prediction_true = prediction_true.astype(np.uint8)