o e0@s.ddlZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z ddl mZddlmZddlmZddlmZd d Zd d Zd dZddZddZddZddZddZddZddZddZdd Z d!d"Z!d#d$Z"d%d&Z#d'd(Z$d)d*Z%d+d,Z&d-d.Z'e(d/kre'dSdS)0N)config)rfapi)APP_URLDEFAULT_BATCH_NAME&get_conditional_configuration_variableload_roboflow_api_key)ClassificationModel)InstanceSegmentationModel)ObjectDetectionModel)SemanticSegmentationModelcCs tdS)N)roboflowlogin)argsrB/var/www/myenv/lib/python3.10/site-packages/roboflow/roboflowpy.pyr s r cCs^d}t||}|r-|dp|d}|dp|d}|dp'|d}|||fSdS) Nzx(?:https?://)?(?:universe|app)\.roboflow\.(?:com|one)/([^/]+)/([^/]+)(?:/dataset)?(?:/(\d+))?|([^/]+)/([^/]+)(?:/(\d+))?)NNN)rematchgroup)urlregexr organizationdatasetversionrrr _parse_urls  rcCst}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)r Roboflowr datasetUrl workspaceprojectversionsprintexitrintdownloadformatr")rrfwpvr'r(rrrrr,sr,cCs2t}||j}|j|j|j|j|jddS)N) dataset_pathdataset_format project_name num_workers)r r$r&upload_datasetfolderr-r' concurrency)rr.r&rrrimport_dataset/s  r9c Cs^t}||j}||j}|j|j|j|j|j|j |j |j r'|j dng|j ddS)N,) image_pathannotation_pathannotation_labelmapsplitnum_retry_uploads batch_name tag_names is_prediction) r r$r&r' single_upload imagefile annotationlabelmapr> num_retriesbatchrArB)rr.r&r'rrr upload_image:s   rIcCst}||j}|j}|D]F}tt|dtdtd|dtd|dtd|dtd|d td |d td |d qdS)Nname link: /id id: z type: typez versions: r(z images: imagesz classes: classes)r r$r& project_listr)rkeys)rr.r&projectsr0rrr list_projectsJs  rUcCsxtj}tddd}|D],}tt|d|d|krdndtdtd|dtd |dq dS) N RF_WORKSPACEdefaultrJrz (default workspace)rKrLrN)roboflow_config RF_WORKSPACESvaluesrr)r)r workspaces rf_workspacer/rrrlist_workspacesYs  $r_cCs.t|j}t||j}ttj|dddS)Nrindent)r workspaceIdr get_workspacer)jsondumps)rapi_keyworkspace_jsonrrrrccs rccCs@|jptddd}t|}t|||j}ttj|dddS)NrVrWrr`) r&rrr get_project projectIdr)rdre)r workspace_urlrf dataset_jsonrrrrhisrhc Cs|jptddd}t|}|d|j}|j}|s/|jd\}}t|||}|dd}tt t t d|}|||} i} |j durQ|dvrQt |j d| d <|jdurc|d krct |jd| d <| j|jfi| } t| dS) NrVrWrLr'rOobject-detectionclassificationinstance-segmentationsemantic-segmentation)rmrorpd confidencermoverlap)r&rrmodelrOr>rrhr rr r rrr+rspredictfiler)) rrjrf project_url projectTyperi_rk modelClassrtkwargsrrrrinferps.   r|cCsTtjdd}|jdd}t|t|t|t|t|t|t ||S)NuDWelcome to the roboflow CLI: computer vision at your fingertips 🪄) description subcommandstitle) argparseArgumentParseradd_subparsers_add_login_parser_add_download_parser_add_upload_parser_add_import_parser_add_infer_parser_add_projects_parser_add_workspaces_parser)parser subparsersrrr _argparsers  rcCsN|jddd}|jddd|jdddd d |jd d d d|jtddS)Nr,zDDownload a dataset version from your workspace or Roboflow Universe.helpr%z0Dataset URL (e.g., `roboflow-100/cells-uyemf/2`)-fr-voczSpecify 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])destrXrz-lr"z Location to download the datasetrrfunc) add_parser add_argument set_defaultsr,)rdownload_parserrrrrs rcCs|jddd}|jddd|jdddd |jd d d d |jd ddd |jdddd |jddddd|jdddtdd|jdddtd|jd d!d"d |jd#d$d%d&d'|jtd(dS))Nuploadz"Upload a single image to a datasetrrDzpath to image file-wr&Kspecify a workspace url or id (will use default workspace if not specified)r-pr'z#project_id to upload the image intoz-arEz"path to annotation file (optional)-mrFz path to labelmap file (optional)z-sr>z)split set (train, valid, test) - optionaltrainrrrXz-rrGz1Retry failed uploads this many times (default: 0)r)rrrOrXz-brHz"Batch name to upload to (optional)-trAz*Tag names to apply to the image (optional)z-irBz.Whether this upload is a prediction (optional) store_true)rractionr)rrr+rrrI)r upload_parserrrrrsprcCsr|jddd}|jddd|jdddd |jd d d d |jd dtddd|jddddd|jtddS)Nimportz$Import a dataset from a local folderrr7z6filesystem path to a folder that contains your datasetrr&rrrr'z,project will be created if it does not exist-cr8zsB          ; 1