添加了qt传输模型名称的功能

This commit is contained in:
FEIJINTI 2023-03-06 16:14:25 +08:00
parent 363443ae94
commit bbcc69d48e

View File

@ -35,7 +35,7 @@ def process_cmd(cmd: str, data: any, connected_sock: socket.socket, detector: Wo
detector = WoodClass(w=4096, h=1200, n=3000, debug_mode=False) detector = WoodClass(w=4096, h=1200, n=3000, debug_mode=False)
model_name = None model_name = None
if "/n" in data: if "/n" in data:
data, model_name = data.split("/n", 1) data, model_name = data.split(":", 1)
model_name = model_name + ".p" model_name = model_name + ".p"
settings.data_path = data settings.data_path = data
settings.model_path = ROOT_DIR / 'models' / detector.fit_pictures(data_path=settings.data_path, file_name=model_name) settings.model_path = ROOT_DIR / 'models' / detector.fit_pictures(data_path=settings.data_path, file_name=model_name)