From ae54b4f704e5d90b3c34f272bb945b251e5aeef2 Mon Sep 17 00:00:00 2001 From: somunslotus Date: Thu, 11 Jul 2024 09:45:25 +0800 Subject: [PATCH] =?UTF-8?q?=E7=BB=93=E6=9E=9C=E5=8F=AF=E8=A7=86=E5=8C=96?= =?UTF-8?q?=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- msunet/main.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/msunet/main.py b/msunet/main.py index e9ad124..ab25c25 100644 --- a/msunet/main.py +++ b/msunet/main.py @@ -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)