Update main.py

ZhanYang-nwpu · Apr 10, 2023 · c653b78 · c653b78
1 parent 16a7312
commit c653b78
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -36,9 +36,9 @@
 def main():
     parser = argparse.ArgumentParser(description='Dataloader test')
     parser.add_argument('--size', default=640, type=int, help='image size')
-    parser.add_argument('--images_path', type=str, default='DIOR_RSVGD\\JPEGImages',
+    parser.add_argument('--images_path', type=str, default='DIOR_RSVG\\JPEGImages',
                         help='path to dataset splits data folder')
-    parser.add_argument('--anno_path', type=str, default='DIOR_RSVGD\\Annotations',
+    parser.add_argument('--anno_path', type=str, default='DIOR_RSVG\\Annotations',
                         help='location of pre-parsed dataset info')
     parser.add_argument('--time', default=40, type=int,
                         help='maximum time steps (lang length) per batch')
@@ -538,4 +538,4 @@ def Mytest_epoch(test_loader, model, mode='test'):
     logging.info(final_str)
 
 if __name__ == "__main__":
-    main()
+    main()