Henry Scheible commited on
Commit
a064e53
1 Parent(s): a942e95

add pre encoding step

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -127,6 +127,7 @@ def count_barnacles(image_raw, split_num, progress=gr.Progress()):
127
  # plt.figure()
128
  # plt.imshow(small_image)
129
  # plt.axis('on')
 
130
  progress(0, desc=f"Generating masks for crop {r*split_num + c}/{split_num ** 2}")
131
  masks = mask_generator.generate(small_image)
132
  num_masks = len(masks)
 
127
  # plt.figure()
128
  # plt.imshow(small_image)
129
  # plt.axis('on')
130
+ mask_generator.predictor.set_image(small_image)
131
  progress(0, desc=f"Generating masks for crop {r*split_num + c}/{split_num ** 2}")
132
  masks = mask_generator.generate(small_image)
133
  num_masks = len(masks)