o ĎiSO@sddlZddlZddlZddlZddlmZddlmZddlmZddl m Z m Z m Z ddl mZddlmZddlmZdd lmZdd lmZd d Zd dZddZddZddZddZddZddZddZddZ dd Z!d!d"Z"d#d$Z#d%d&Z$d'd(Z%d)d*Z&d+d,Z'd-d.Z(d/d0Z)d1d2Z*d3d4Z+d5d6Z,d7d8Z-d9d:Z.d;d<Z/d=d>Z0d?d@Z1dAdBZ2dCdDZ3dEdFZ4e5dGkre4dSdS)HN)config) deployment)rfapi)APP_URL&get_conditional_configuration_variableload_roboflow_api_key)ClassificationModel)InstanceSegmentationModel)KeypointDetectionModel)ObjectDetectionModel)SemanticSegmentationModelcCstj|jddS)N)force)roboflowloginr argsrO/home/jeff/fluffinator/venv/lib/python3.10/site-packages/roboflow/roboflowpy.pyrsrcCsJt}||j}||j}||j}|j|j|jd}t |dS)N) model_type checkpoint) rRoboflow workspaceprojectversionversion_numbertrainrrprint)rrfrrrmodelrrrrs     rcCs^d}t||}|r-|dp|d}|dp|d}|dp'|d}|||fSdS) Nzx(?:https?://)?(?:universe|app)\.roboflow\.(?:com|one)/([^/]+)/([^/]+)(?:/dataset)?(?:/(\d+))?|([^/]+)/([^/]+)(?:/(\d+))?)NNN)rematchgroup)urlregexr& organizationdatasetrrrr _parse_urls  r,cCst}t|j\}}}|||}|s6|}|s(td|dtd|d}td|j dn| t |}|j |j |j dddS) Nzproject z# does not have any version. exitingrz,Version not provided. Downloading last one ()T)location overwrite)rrr, datasetUrlrrversionsrexitrintdownloadformatr/)rrwpvrr2rrrrr5*sr5cCsBt|j}t|}||j}|j|j|j|j|j|j ddS)N) dataset_path project_name num_workers batch_name num_retries) rrrrupload_datasetfolderr concurrencyr=r>)rapi_keyrrrrrimport_dataset:s    rCc Cs^t}||j}||j}|j|j|j|j|j|j |j |j r'|j dng|j ddS)N,) image_pathannotation_pathannotation_labelmapsplitnum_retry_uploadsr= tag_names is_prediction) rrrr single_upload imagefile annotationlabelmaprHr>batchrJrK)rrrrrrr upload_imageGs   rQcCst|j}||j}|jdurszshow_whoami..riz **********apiKeyr!rs)rrogetrrwrx)rrnworkspaces_by_urldefault_workspace_urldefault_workspacerrr show_whoamis   rcCs|jddd}|jtddS)Nwhoamizshow current user inforr) add_parserrrrdownload_parserrrrrsrcCsN|jddd}|jddd|jdddd d |jd d d d|jtddS)Nr5zDDownload a dataset version from your workspace or Roboflow Universe.rr1z0Dataset URL (e.g., `roboflow-100/cells-uyemf/2`)-fr6voczSpecify the format to download the version. Available options: [coco, yolov5pytorch, yolov7pytorch, my-yolov6, darknet, voc, tfrecord, createml, clip, multiclass, coco-segmentation, yolo5-obb, png-mask-semantic, yolov8, yolov9]destrkrz-lr/z Location to download the datasetrrr)rrrr5rrrrrs rcCs|jddd}|jddd|jdddd |jd d d d |jd ddd |jdddd |jddddd|jdddtdd|jdddd |jd d!d"d |jd#d$d%d&d'|jtd(dS))Nuploadz"Upload a single image to a datasetrrMzpath to image file-wrKspecify a workspace url or id (will use default workspace if not specified)r-pr#project_id to upload the image into-arNz"path to annotation file (optional)-mrOz path to labelmap file (optional)z-srHz)split set (train, valid, test) - optionalr)rrrk-rr>z1Retry failed uploads this many times (default: 0)r)rrrbrkz-brPz"Batch name to upload to (optional)-trJz*Tag names to apply to the image (optional)z-irKz.Whether this upload is a prediction (optional)rrrrr)rrr4rrQ)r upload_parserrrrr snrcCsp|jddd}|jdddd|jdd d d|jd d td d|jdddd|jdddd|jtddS)Nrz#Train a model for a dataset versionrrrrrrrz!project_id to train the model forrrzversion number to trainrrbrrrz7type of the model to train (e.g., rfdetr-nano, yolov8n)z --checkpointrz"checkpoint to resume training fromr)rrr4rr)r train_parserrrrrCs8rcCs|jddd}|jddd|jdddd |jd d d d |jd dtddd|jdddd |jddtddd|jtddS)Nimportz$Import a dataset from a local folderrr@z6filesystem path to a folder that contains your datasetrrrrrrz,project will be created if it does not exist-crAzz0Retry failed uploads this many times (default=0)rr)rrr4rrC)r import_parserrrrrbs> rcCs|jddd}|jdd}|jddd}|jdd d d |jtd |jd dd}|jddd|jdd d d |jtd dS)NrzOproject related commands. type 'roboflow project' to see detailed command helprzproject subcommandsrrWz list projectsrrrrrrz show detailed info for a projectr project ID)rrrrrhr)rproject_parserprojectsubparsersprojectlist_parserprojectget_parserrrrrs.  rcCs`|jddd}|jdd}|jddd}|jtd|jd d d}|jd d d|jtddS) NrzSworkspace related commands. type 'roboflow workspace' to see detailed command helprzworkspace subcommandsrrWzlist workspacesrrz"show detailed info for a workspacerur)rrrrrrrv)rworkspace_parserworkspacesubparsersworkspacelist_parserworkspaceget_parserrrrrs  rcCst|jddd}|jdddd|jddd d|jd d td d |jdddd|jddtddd|jtddS)Nrzrun video inference apirrrBrrrrrr%version number to upload the model torrr|zpath to video filez-fpsr}r"rr)rrr4rr)rrun_video_inference_api_parserrrrrsBrcCs|jddd}|jddd|jdddd |jd d d d |jd dtddd|jddtddd|jdddgdd|jtddS)Nrzperform inference on an imagerrz filesystem path to an image filerrrrrrz?model id (id of a version with trained model e.g. my-project/3)rrzkspecify a confidence threshold between 0.0 and 1.0, default is 0.5(only applies to object-detection models)g?rz-orzispecify an overlap threshold between 0.0 and 1.0, default is 0.5(only applies to object-detection models)rrbz5specify the model type to skip api call to look it up)rrrr)rrchoicesr)rrfloatrr)r infer_parserrrrrsN rcCs|jddd}|jdddd|jddd d|jd d d d d|jddtddd|jdddd|jdddd|jddddd|jddd d|jtd!dS)"Nr\z"Upload a trained model to RoboflowrrrBrrrrrrappendzEproject_id to upload the model into (can be specified multiple times))rrrrrz0version number to upload the model to (optional)rrrz(type of the model (e.g., yolov8, yolov5)rrRzpath to the trained model filerrUzweights/best.ptzname of the model filerrrTzname of the modelr)rrr4rr\)rupload_model_parserrrrrsbrcCs`|jddd}|jdddd|jddd d|jd d d d|jd dtdd|jtddS)Nget_workspace_infoz"get workspace project version inforrrBrrrrrrz#project_id to upload the model intorrrrr)rrr4rr)r workspace_project_version_parserrrrr8s4rcCs0|jddd}|jddddd|jtd dS) NrzLog in to Roboflowrrr z force loginrrr)rrrr)r login_parserrrrrUsrcCs2t}|}t|dr||dS|dS)Nr)r parse_argshasattrr print_help)rrrrrmain`s   r__main__)6rrwr%rrrmrroboflow.adaptersrroboflow.configrrrroboflow.models.classificationr%roboflow.models.instance_segmentationr "roboflow.models.keypoint_detectionr roboflow.models.object_detectionr %roboflow.models.semantic_segmentationr rrr,r5rCrQr\rhrrrvrrrrrrrrrrrrrrrrrrrr__name__rrrrs\              :"%14