Browse Source

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 years ago
parent
commit
f94f1ba931
1 changed files with 1 additions and 1 deletions
  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))