Henry Scheible
commited on
Commit
•
a064e53
1
Parent(s):
a942e95
add pre encoding step
Browse files
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)
|