tbuyuktanir commited on
Commit
5b6ef6a
·
verified ·
1 Parent(s): 829b414

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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, _, _, _ = model.eval(img_np, diameter=None if diameter == 0 else diameter, channels=[0, 0])
 
 
 
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