Update inference.py

LBH1024 · Jul 23, 2022 · 1a3e08c · 1a3e08c
1 parent c1928a1
commit 1a3e08c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/inference.py b/inference.py
@@ -14,9 +14,9 @@
 
 parser = argparse.ArgumentParser(description='model testing')
 parser.add_argument('--dataset', default='CROHME', type=str, help='数据集名称')
-parser.add_argument('--image_path', default='dataset/CROHME/14_test_images.pkl', type=str, help='测试image路径')
-parser.add_argument('--label_path', default='dataset/CROHME/14_test_labels.txt', type=str, help='测试label路径')
-parser.add_argument('--word_path', default='dataset/CROHME/words_dict.txt', type=str, help='测试dict路径')
+parser.add_argument('--image_path', default='datasets/CROHME/14_test_images.pkl', type=str, help='测试image路径')
+parser.add_argument('--label_path', default='datasets/CROHME/14_test_labels.txt', type=str, help='测试label路径')
+parser.add_argument('--word_path', default='datasets/CROHME/words_dict.txt', type=str, help='测试dict路径')
 
 parser.add_argument('--draw_map', default=False)
 args = parser.parse_args()