修改模型存放路径
This commit is contained in:
parent
4f441f57f8
commit
2d4ba01bf0
|
@ -6,3 +6,10 @@
|
|||
# Datasource local storage ignored files
|
||||
/dataSources/
|
||||
/dataSources.local.xml
|
||||
|
||||
*.iml
|
||||
*.ipr
|
||||
*.iws
|
||||
*.idea/workspace.xml
|
||||
*.idea/modules.xml
|
||||
*.xml
|
||||
|
|
|
@ -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(
|
||||
|
|
Loading…
Reference in New Issue