@@ -21,30 +21,29 @@ def upload(args):
2121 project_name = p ,
2222 )
2323
24-
25- def main ():
24+ def _argparser ():
2625 parser = argparse .ArgumentParser (description = "main description" )
2726 subparsers = parser .add_subparsers (title = "subcommands" )
28-
2927 login_parser = subparsers .add_parser ("login" , help = "Log in to Roboflow" )
3028 login_parser .set_defaults (func = login )
31-
3229 download_parser = subparsers .add_parser ("download" , help = "Download a dataset version from your workspace or Roboflow Universe." )
3330 download_parser .add_argument ("datasetUrl" , help = "Dataset URL (e.g., `roboflow-100/cells-uyemf/2`)" )
3431 download_parser .add_argument ("-f" ,
3532 choices = ["coco" , "yolov5pytorch" , "yolov7pytorch" , "my-yolov6" , "darknet" , "voc" , "tfrecord" ,
36- "createml" , "clip" , "multiclass" , "coco-segmentation" , "yolo5-obb" , "png-mask-semantic" ],
33+ "createml" , "clip" , "multiclass" , "coco-segmentation" , "yolo5-obb" , "png-mask-semantic" , "yolov8" ],
3734 help = "Specify the format to download the version in (default: interactive prompt)" )
3835 download_parser .add_argument ("-l" , help = "Location to download the dataset" )
3936 download_parser .set_defaults (func = download )
40-
4137 upload_parser = subparsers .add_parser ("upload" , help = "Upload a dataset" )
4238 upload_parser .add_argument ("folder" , help = "filesystem path to a folder that contains your dataset" )
4339 upload_parser .add_argument ("-w" , help = "workspace url" )
4440 upload_parser .add_argument ("-p" , help = "Project name" )
4541 upload_parser .add_argument ("-f" , choices = ["voc" , "yolov8" , "yolov5" ], help = "format" )
4642 upload_parser .set_defaults (func = upload )
43+ return parser
4744
45+ def main ():
46+ parser = _argparser ()
4847 args = parser .parse_args ()
4948 if hasattr (args , 'func' ):
5049 args .func (args )
0 commit comments