自动新建输出文件夹
This commit is contained in:
parent
2b45bd6d60
commit
d5ba9b2cf4
@ -60,6 +60,7 @@ def main(opt):
|
|||||||
result = SA_Detect(model_path, data, opt.batch_size)
|
result = SA_Detect(model_path, data, opt.batch_size)
|
||||||
# result.to_csv(Path(opt.output) / (one_file.stem + "segment.csv"), index=False)
|
# result.to_csv(Path(opt.output) / (one_file.stem + "segment.csv"), index=False)
|
||||||
if opt.output:
|
if opt.output:
|
||||||
|
Path(opt.output).mkdir(parents=True, exist_ok=True)
|
||||||
save_path = Path(opt.output) / (one_file.stem + ".csv")
|
save_path = Path(opt.output) / (one_file.stem + ".csv")
|
||||||
else:
|
else:
|
||||||
save_path = None
|
save_path = None
|
||||||
|
Loading…
Reference in New Issue
Block a user