cloverta/cellpose-web
1
0
Fork 0
mirror of https://github.com/ClovertaTheTrilobita/cellpose-web.git synced 2026-04-01 23:14:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feature(config): 新增模型自定义路径

Browse source
This commit is contained in:
ClovertaTheTrilobita 2025-09-22 17:47:16 +00:00
parent 04e2bf0edc
commit 98351dc096
2 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
backend/config.yaml
View file

@ -1,4 +1,5 @@
model:
save_dir: models
data:
root_dir: .

11
backend/cp_run.py
View file

@ -1,8 +1,7 @@
from cellpose import models, plot
from cellpose.io import imread, save_masks
import os
from PIL import Image
import numpy as np
import os, datetime
import datetime
import time
from omegaconf import OmegaConf
from pathlib import Path
@ -14,6 +13,12 @@ BASE_DIR = cfg.data.root_dir
UPLOAD_DIR = cfg.data.upload_dir
OUTPUT_DIR = cfg.data.run.output_dir
OUTPUT_TEST_DIR = cfg.data.run.test_output_dir
MODELS_DIR = str((CONFIG_PATH.parent / cfg.model.save_dir).resolve())
os.makedirs(MODELS_DIR, exist_ok=True)
os.environ["CELLPOSE_LOCAL_MODELS_PATH"] = MODELS_DIR
from cellpose import models, plot
from cellpose.io import imread, save_masks
class Cprun: