修改模型存放路径

This commit is contained in:
somunslotus 2024-07-11 09:06:15 +08:00
parent 4f441f57f8
commit 2d4ba01bf0
3 changed files with 10 additions and 1 deletions

7
.idea/.gitignore vendored
View File

@ -6,3 +6,10 @@
# Datasource local storage ignored files
/dataSources/
/dataSources.local.xml
*.iml
*.ipr
*.iws
*.idea/workspace.xml
*.idea/modules.xml
*.xml

View File

@ -190,7 +190,7 @@ def main(opts: GlobalOptions) -> int:
files = np.array(glob.glob(save_path["pre_process_img"] + "/*.png")).tolist()
if opts.model_path is None:
model_path = './result/train/last.ckpt'
model_path = './model/last.ckpt'
else:
model_path = opts.model_path.get_full_path()
filename = predict(model_path, files, save_path["predict"])
@ -212,6 +212,8 @@ def main(opts: GlobalOptions) -> int:
report = Report(name="预测结果",title="原子定位结果", sections=[])
report.save(opts.output_dir)
def generate_report(save_path: Dict[str, str]) -> None:
pass
def to_parser():
return to_runner(