Fixed argparse error

RapidAI · Oct 24, 2023 · ad6caf0 · ad6caf0
1 parent 7ba5b6b
commit ad6caf0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/label_convert/labelImg_to_publaynet.py b/label_convert/labelImg_to_publaynet.py
@@ -192,7 +192,7 @@ def main():
         "--data_dir", type=str, required=True, help="The directory from labelImg."
     )
     parser.add_argument("--val_ratio", type=float, default=0.2)
-    parser.add_argument("--have_test", type=bool, default=True)
+    parser.add_argument("--have_test", action="store_true", default=False)
     parser.add_argument("--test_ratio", type=float, default=0.2)
     args = parser.parse_args()
 

diff --git a/label_convert/labelImg_to_yolov5.py b/label_convert/labelImg_to_yolov5.py
@@ -135,7 +135,7 @@ def main():
     parser.add_argument("--src_dir", type=str)
     parser.add_argument("--out_dir", type=str)
     parser.add_argument("--val_ratio", type=float, default=0.2)
-    parser.add_argument("--have_test", type=bool, default=True)
+    parser.add_argument("--have_test", action="store_true", default=False)
     parser.add_argument("--test_ratio", type=float, default=0.2)
     args = parser.parse_args()