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); };