somunslotus
/
atom-predict
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

结果可视化修改

This commit is contained in:
somunslotus 2024-07-11 09:45:25 +08:00
parent 82586663c3
commit ae54b4f704
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
msunet/main.py
View File

@ -238,17 +238,16 @@ def generate_report(save_path: Dict[str, str], output_dir: str) -> None:
img_elements = []
#预测结果在遍历,
for img_path in glob.glob(save_path["post_process"] + "/*.png"):
for img_path in glob.glob(save_path["plot_view"] + "/*.png"):
img_elements.append(AutoReportElement(
path=img_path,
title=img_path.split('/')[-1],
description=f'预测结果',
) )
post_process_img_section = ReportSection(title="预测结果", ncols=3, elements=img_elements)
report = Report(name="预测结果",title="原子定位结果", sections=[ori_img_section, pre_process_img_section, post_process_img_section])
report = Report(name="预测结果",title="原子定位结果", sections=[ post_process_img_section, ori_img_section, pre_process_img_section])
report.save(output_dir)