From e4eb06d71c624f268195170d8a708e34170dba35 Mon Sep 17 00:00:00 2001 From: mozzie Date: Fri, 20 Sep 2024 16:29:55 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BC=98=E5=8C=96=E7=AE=97=E6=B3=95?= =?UTF-8?q?=E5=88=86=E5=89=B2=E8=BE=93=E5=87=BA=E6=96=87=E4=BB=B6=E5=A4=B9?= =?UTF-8?q?=E7=BB=93=E6=9E=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/desktop/electron/ipcEvent/common/index.ts | 8 ++++++-- apps/desktop/src/pages/Boot/index.tsx | 4 ++-- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/apps/desktop/electron/ipcEvent/common/index.ts b/apps/desktop/electron/ipcEvent/common/index.ts index f922011..fb3e477 100644 --- a/apps/desktop/electron/ipcEvent/common/index.ts +++ b/apps/desktop/electron/ipcEvent/common/index.ts @@ -36,15 +36,19 @@ export const registerCommonHandler = () => { ipcMain.handle("model:infer", async (_event, SeriesInstanceUIDs) => { // 构造推理任务参数列表 - const save_path = path.join(app.getPath("userData"), "output"); const pu = InferDeviceEnum.GPU; - const module = InferStructuralEnum.AORTA; + const module = InferStructuralEnum.PERI; const turbo = true; const seg_schedule = true; for (let i = 0; i < SeriesInstanceUIDs.length; i++) { const SeriesInstanceUID = SeriesInstanceUIDs[i]; // 下载dicom到本地，获取文件夹路径 const img_path = await downloadSeriesDicomFiles(SeriesInstanceUID); + const save_path = path.join( + app.getPath("userData"), + "output", + SeriesInstanceUID + ); const task = { save_path, pu, module, turbo, seg_schedule, img_path }; console.log(task); const result = await executeInferTask(task, (data) => { diff --git a/apps/desktop/src/pages/Boot/index.tsx b/apps/desktop/src/pages/Boot/index.tsx index dd513a1..4786db0 100644 --- a/apps/desktop/src/pages/Boot/index.tsx +++ b/apps/desktop/src/pages/Boot/index.tsx @@ -27,8 +27,8 @@ const Boot = () => { */ const handleTasks = () => { const SeriesInstanceUIDs = [ - // "1.2.156.112605.14038010222575.230518044041.3.4344.300061", - "1.3.12.2.1107.5.1.4.73399.30000020080900171669200001479", + "1.2.156.112605.14038010222575.230518044041.3.4344.300061", + // "1.3.12.2.1107.5.1.4.73399.30000020080900171669200001479", ]; window.ipcRenderer.invoke("model:infer", SeriesInstanceUIDs); };