瀏覽代碼

fix to use the script pass-in args save_dir

Traceback (most recent call last):
  File "compare.py", line 160, in <module>
    format(save_dir))
NameError: name 'save_dir' is not defined
bian.xie 4 年之前
父節點
當前提交
f94f1ba931
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      examples/industrial_quality_inspection/compare.py

+ 1 - 1
examples/industrial_quality_inspection/compare.py

@@ -157,4 +157,4 @@ if __name__ == '__main__':
 
     print(
         "The visualized ground-truths and predictions are saved in {}. Ground-truth is on the left, prediciton is on the right".
-        format(save_dir))
+        format(args.save_dir))