Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -15,7 +15,10 @@ def masks_to_overlay(image, masks):
|
|
| 15 |
def segment_and_count(image: Image.Image, model_type: str, diameter: float):
|
| 16 |
img_np = np.array(image.convert("RGB"))
|
| 17 |
model = models.CellposeModel(model_type=model_type)
|
| 18 |
-
masks, _, _
|
|
|
|
|
|
|
|
|
|
| 19 |
|
| 20 |
overlay = masks_to_overlay(img_np, masks)
|
| 21 |
cell_count = len(np.unique(masks)) - 1 # exclude background 0
|
|
|
|
| 15 |
def segment_and_count(image: Image.Image, model_type: str, diameter: float):
|
| 16 |
img_np = np.array(image.convert("RGB"))
|
| 17 |
model = models.CellposeModel(model_type=model_type)
|
| 18 |
+
masks, _, _ = model.eval(
|
| 19 |
+
img_np,
|
| 20 |
+
diameter=None if diameter == 0 else diameter,
|
| 21 |
+
channels=[0, 0])
|
| 22 |
|
| 23 |
overlay = masks_to_overlay(img_np, masks)
|
| 24 |
cell_count = len(np.unique(masks)) - 1 # exclude background 0
|