Hugging Face's logo

Spaces:
KalbeDigitalLab
/
medsam-segment-anything
Running

App Files Community
haritsahm commited on
Commit
ef9685e
·
1 Parent(s): 84be203

minor fix to variable name

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -18,7 +18,7 @@ def process_box(predictor_model, show_mask, radius_width):
18
  width, height = bg_image.size[:2]
19
  container_width = 700
20
  scale = container_width/width
21
- scaled_hw = (container_width, int(height * scale))
22
 
23
  if not predictor_model.set_image:
24
  np_image = np.asanyarray(bg_image)
@@ -26,7 +26,7 @@ def process_box(predictor_model, show_mask, radius_width):
26
  predictor_model.set_image(np_image)
27
 
28
  if 'result_image' not in st.session_state:
29
- st.session_state.result_image = bg_image.resize(scaled_hw)
30
 
31
  box_canvas = st_canvas(
32
  fill_color="rgba(255, 255, 0, 0)",
@@ -82,7 +82,7 @@ def process_box(predictor_model, show_mask, radius_width):
82
  im_masked = utils.show_click(masks,color)
83
  im_masked = Image.fromarray(im_masked).convert('RGBA')
84
  result_image = Image.alpha_composite(Image.fromarray(bg_image).convert('RGBA'),im_masked).convert("RGB")
85
- result_image = result_image.resize(scaled_hw)
86
  st.session_state.display_result = True
87
  return result_image
88
  else:
@@ -97,7 +97,7 @@ def process_click(predictor_model, show_mask, radius_width):
97
  width, height = bg_image.size[:2]
98
  container_width = 700
99
  scale = container_width/width
100
- scaled_hw = (container_width, int(height * scale))
101
 
102
  if not predictor_model.set_image:
103
  np_image = np.asanyarray(bg_image)
@@ -105,7 +105,7 @@ def process_click(predictor_model, show_mask, radius_width):
105
  predictor_model.set_image(np_image)
106
 
107
  if 'result_image' not in st.session_state:
108
- st.session_state.result_image = bg_image.resize(scaled_hw)
109
 
110
  click_canvas = st_canvas(
111
  fill_color="rgba(255, 255, 0, 0.8)",
@@ -160,7 +160,7 @@ def process_click(predictor_model, show_mask, radius_width):
160
  im_masked = utils.show_click(masks,color)
161
  im_masked = Image.fromarray(im_masked).convert('RGBA')
162
  result_image = Image.alpha_composite(Image.fromarray(bg_image).convert('RGBA'),im_masked).convert("RGB")
163
- result_image = result_image.resize(scaled_hw)
164
  st.session_state.display_result = True
165
  return result_image
166
  else:
@@ -174,10 +174,10 @@ def process_everything(automask_model, show_mask, radius_width):
174
  width, height = bg_image.size[:2]
175
  container_width = 700
176
  scale = container_width/width
177
- scaled_hw = (container_width, int(height * scale))
178
 
179
  if 'result_image' not in st.session_state:
180
- st.session_state.result_image = bg_image.resize(scaled_hw)
181
 
182
  dummy_canvas = st_canvas(
183
  fill_color="rgba(255, 255, 0, 0.8)",
@@ -210,7 +210,7 @@ def process_everything(automask_model, show_mask, radius_width):
210
  im_masked = utils.show_everything(masks)
211
  im_masked = Image.fromarray(im_masked).convert('RGBA')
212
  result_image = Image.alpha_composite(Image.fromarray(bg_image).convert('RGBA'),im_masked).convert("RGB")
213
- result_image = result_image.resize(scaled_hw)
214
  st.session_state.display_result = True
215
  return result_image
216
 
 
18
  width, height = bg_image.size[:2]
19
  container_width = 700
20
  scale = container_width/width
21
+ scaled_wh = (container_width, int(height * scale))
22
 
23
  if not predictor_model.set_image:
24
  np_image = np.asanyarray(bg_image)
 
26
  predictor_model.set_image(np_image)
27
 
28
  if 'result_image' not in st.session_state:
29
+ st.session_state.result_image = bg_image.resize(scaled_wh)
30
 
31
  box_canvas = st_canvas(
32
  fill_color="rgba(255, 255, 0, 0)",
 
82
  im_masked = utils.show_click(masks,color)
83
  im_masked = Image.fromarray(im_masked).convert('RGBA')
84
  result_image = Image.alpha_composite(Image.fromarray(bg_image).convert('RGBA'),im_masked).convert("RGB")
85
+ result_image = result_image.resize(scaled_wh)
86
  st.session_state.display_result = True
87
  return result_image
88
  else:
 
97
  width, height = bg_image.size[:2]
98
  container_width = 700
99
  scale = container_width/width
100
+ scaled_wh = (container_width, int(height * scale))
101
 
102
  if not predictor_model.set_image:
103
  np_image = np.asanyarray(bg_image)
 
105
  predictor_model.set_image(np_image)
106
 
107
  if 'result_image' not in st.session_state:
108
+ st.session_state.result_image = bg_image.resize(scaled_wh)
109
 
110
  click_canvas = st_canvas(
111
  fill_color="rgba(255, 255, 0, 0.8)",
 
160
  im_masked = utils.show_click(masks,color)
161
  im_masked = Image.fromarray(im_masked).convert('RGBA')
162
  result_image = Image.alpha_composite(Image.fromarray(bg_image).convert('RGBA'),im_masked).convert("RGB")
163
+ result_image = result_image.resize(scaled_wh)
164
  st.session_state.display_result = True
165
  return result_image
166
  else:
 
174
  width, height = bg_image.size[:2]
175
  container_width = 700
176
  scale = container_width/width
177
+ scaled_wh = (container_width, int(height * scale))
178
 
179
  if 'result_image' not in st.session_state:
180
+ st.session_state.result_image = bg_image.resize(scaled_wh)
181
 
182
  dummy_canvas = st_canvas(
183
  fill_color="rgba(255, 255, 0, 0.8)",
 
210
  im_masked = utils.show_everything(masks)
211
  im_masked = Image.fromarray(im_masked).convert('RGBA')
212
  result_image = Image.alpha_composite(Image.fromarray(bg_image).convert('RGBA'),im_masked).convert("RGB")
213
+ result_image = result_image.resize(scaled_wh)
214
  st.session_state.display_result = True
215
  return result_image
216