o Ďi? @sddlZddlZddlZddlZddlmZddlZddlmZde de de de fdd Z de defd d Z de de de de fd d Z de de de de fddZ de de fddZde fddZ dde de de de fddZdde de de de fddZdS)N)Callable)*print_warn_for_wrong_dependencies_versions model_type model_pathfilenamereturncCst|}||||SN)_get_processor_function)rrr processorr Y/home/jeff/fluffinator/venv/lib/python3.10/site-packages/roboflow/util/model_processor.pyprocess s r csgd}tfdd|Dstdd|dr#z*_get_processor_function.. Model type z% not supported. Supported models are )rrrc3rrr )rmodelrr r r'r ) zflorence-2-basezflorence-2-largezpaligemma-3b-pt-224zpaligemma-3b-pt-448zpaligemma-3b-pt-896zpaligemma2-3b-pt-224zpaligemma2-3b-pt-448zpaligemma2-3b-pt-896zpaligemma2-3b-pt-224-peftzpaligemma2-3b-pt-448-peftzpaligemma2-3b-pt-896-peftzB not supported for this type of upload.Supported upload types are rr)any ValueError startswith RuntimeError_process_huggingface_process_yolonas_process_rfdetr _process_yolo)rsupported_modelssupported_hf_typesr rr r s$  r c Csd|vr"z ddl}ddl}Wn tytdwtdgddnd|vr=z8.3.0rzThe ultralytics python package is required to deploy yolov12 models. Please install it from `https://github.com/sunsmarterjie/yolov12`z !!! ATTENTION !!! Model must be trained and uploaded using ultralytics from https://github.com/sunsmarterjie/yolov12 or through the Roboflow platform !!! ATTENTION !!! )r-r.z8.3.63rztThe ultralytics python package is required to deploy yolo26 models. Please install it with `pip install ultralytics`F) weights_onlyr"emacSs|dS)Nrr )xr r r sz_process_yolo..)keycSsg|]}|dqS)r )rr2r r r sz!_process_yolo..z-clsnc train_argscS.i|]\}}|dks|dks|dkr||qSr"imgszbatchr rkvalr r r s.z!_process_yolo..)namesyamlr7argsultralytics_versionrcSr9r:r r=r r r r@s(opt.yamlr;img_size batch_size)r;r<rAr7rCrrBmodel_artifacts.jsonw state_dict.pt) results.csv results.pngrIrKroboflow_deploy.ziparcname compress_type)rIrKFile ; not found. Please make sure to provide a valid model path.)%torchr- ImportErrorr&rprintloadospathjoin isinstancerAlist enumerateappendsortr%rBr7rCitems __version__ Exception__dict__open safe_loadhasattrjsondumpsave state_dictzipfileZipFileexistswrite ZIP_DEFLATEDr$)rrrrTr-r"model_instance class_namesir?r7rCmodel_artifactsstreamoptsfp list_files zip_file_namezipMefiler r r r*Es             $                   r*c CsPgd}||vrtd|d|tj|s td|dt|}tdd|Dd}|dur7tdt|||d krPt tj ||tj |d d g}gd }d }t tj ||d :} |D]} | jtj || | t jd qg|D]} tjtj || r| jtj || | t jd qzWd|S1swY|S)N) z rfdetr-basez rfdetr-nanoz rfdetr-smallz rfdetr-mediumz rfdetr-largez rfdetr-xlargezrfdetr-2xlargezrfdetr-seg-nanozrfdetr-seg-smallzrfdetr-seg-mediumzrfdetr-seg-largezrfdetr-seg-xlargezrfdetr-seg-2xlarger!z$ not supported. Supported types are Model path  does not exist.css(|]}|ds|dr|VqdS)z.ptz.pthNendswith)rfr r r rs&z"_process_rfdetr..z4No .pt or .pth model file found in the provided pathz weights.pt)rLrMrIclass_names.txtrNrJrO)r$rXrYrmFileNotFoundErrorlistdirnextr&get_classnames_txt_for_rfdetrshutilcopyrZrkrlrnro) rrr_supported_types model_filespt_filerequired_filesoptional_filesrxryrzr r r r)s6     r)rcCstj|d}tj|rt||Sddl}|jtj||ddd}t|d}d|vrWt|d}|dD] }| |d q8Wdn1sLwYt||St d |d ) NrrcpuF) map_locationr0rCrqrJ z,No class_names.txt file found in model path a. This should only happen on rfdetr models trained before version 1.1.0. Please re-train your model with the latest version of the rfdetr library, or please create a class_names.txt file in the model path with the class names in new lines in the order of the classes in the model. ) rXrYrZrmmaybe_prepend_dummy_classrTrWvarsrdrnr)rrclass_names_pathrTr"rCr class_namer r r r+s$     rclass_name_filecCst| }|}Wdn1swYd}||vrA|d|t|d}||WddS1s:wYdSdS)Nzbackground_class83422 rrJ)rd readlinesinsert writelines)rrrq dummy_classr r r rEs     "r&fine-tuned-paligemma-3b-pt-224.f16.npzc Cstj|std|dt|}td|d|g}tdd|Dd}tdd|Dretd |d gd }|D]}||vrYtd |td }| dkrYt dq@|D]}| |q\n|durwtd|d| |ntd|dt |dkrtd|dtd|ddl } d} | tj|| d} |D]}| jtj|||dqWdn1swYtd| S)Nr{r|zModel files found in z: css|] }|dr|VqdS)z.npzNr}rrzr r r r\sz'_process_huggingface..cSsg|]}|dqS)z .safetensorsr}rr r r r6]sz(_process_huggingface..z9Found .safetensors file in model path. Deploying PyTorch z model.)zpreprocessor_config.jsonzspecial_tokens_map.jsonztokenizer_config.jsonztokenizer.jsonzMissing necessary filezDo you want to continue? (y/n)yr5zFound .npz file z. in model path. Deploying JAX PaliGemma model.z1No .npz or .safetensors file found in model path .rz#No valid files found in model path zZipping files for deploy: zroboflow_deploy.tarrJ)rPz2Uploading to Roboflow... May take several minutes.)rXrYrmrrrVrr#inputlowerexitr^lentarfilerdrZadd) rrrrfiles_to_deploy npz_filenamenecessary_filesrzresr tar_file_nametarr r r r'PsF       r'weights/best.ptcCs$zddl}Wn tytdw|jtj||dd}|dd}tj|d}tj|s>tdtj|d d ttj|d }t |}Wdn1sWwYgd } | D]} | |vrtt|d | d | qb|t |d|vr|dn|d|d|dd|d} ttj|dd} t | | Wdn1swYttj||tj|dgd} d}ttj||d6}| D]*}tjtj||r|jtj|||tjdq|d|fvrtd|dqWd|S1s wY|S)NrziThe torch python package is required to deploy yolonas models. Please install it with `pip install torch`r)rprocessing_paramsrqrEz$You must create an opt.yaml file at z of the format: imgsz: batch_size: architecture: )r;rG architecturez lacks required key z. Required keys: r;rFrGr)r;r<rrHrIrJrK)z results.jsonrMrIrKrNrOrRrS)rTrUr&rWrXrYrZrmrdrBrerrgrhrrrkrlrnror$)rrrrTr"rqopt_pathrtru required_keysr4rsrvrwrxryrzr r r r(sh            r()r)r)rgrXrrktypingrrBroboflow.util.versionsrstrr r r*r)rrr'r(r r r r s2  434  4