mirror of
https://github.com/ClovertaTheTrilobita/cellpose-web.git
synced 2026-04-01 23:14:50 +00:00
16 lines
459 B
Python
16 lines
459 B
Python
|
from cellpose import models
|
||
|
|
from cellpose.io import imread, save_masks
|
||
|
|
|
||
|
|
def test():
|
||
|
|
model = models.CellposeModel(gpu=True)
|
||
|
|
files = ['img.png']
|
||
|
|
imgs = [imread(f) for f in files]
|
||
|
|
masks, flows, styles = model.eval(
|
||
|
|
imgs, flow_threshold=0.4, cellprob_threshold=0.0
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
for img, mask, flow, name in zip(imgs, masks, flows, files):
|
||
|
|
out = name.rsplit('.', 1)[0] + "_output"
|
||
|
|
save_masks(imgs, mask, flow, out, png=True)
|