mozzie/cvpilot-tool
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 优化算法分割输出文件夹结构

Browse Source
This commit is contained in:
mozzie 2024-09-20 16:29:55 +08:00
parent 67b5e488fe
commit e4eb06d71c
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
apps/desktop/electron/ipcEvent/common/index.ts
View File

@ -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) => {

4
apps/desktop/src/pages/Boot/index.tsx
View File

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