a yf@sdddlZddlZddlmZddlmZmZddlmZm Z m Z ddl m Z m Z mZddlmZdde DZd de DZd d Zd d ZejdeddZejdeddZejdeddZejdeddZd+ddZejjejddde d d!fd"d#Zd$d%Zejj ejdeejje d&dejjed'kd(dd)d*Z!dS),N)Image)CUDA_DEVICE_COUNTCUDA_IS_AVAILABLE) TASK2DATA TASK2MODELTASKS)ASSETS WEIGHTS_DIRchecks) TORCH_1_9cCs"g|]}|tt|t|fqS)r rr.0taskr r J/var/www/html/django/DPS/env/lib/python3.9/site-packages/tests/test_cli.py rcCsg|]}tt|qSr )r rr r r rrrcCstj|dddS)z)Execute a shell command using subprocess.T)checkN) subprocessrunsplit)cmdr r rrsrcCs,tdtdtdtdtddS)z?Test various special command-line modes for YOLO functionality.z yolo helpz yolo checksz yolo versionzyolo settings resetzyolo cfgNrr r r rtest_special_modess rztask,model,datacCs td|d|d|ddS)z=Test YOLO training for different tasks, models, and datasets. yolo train  model= data=z imgsz=32 epochs=1 cache=diskNrrmodeldatar r r test_train sr cCs td|d|d|ddS)zWTest YOLO validation process for specified task, model, and data using a shell command.z yolo val rrz imgsz=32 save_txt save_jsonNrrr r rtest_val&sr!cCstd|dtddS)zLTest YOLO prediction on provided sample assets for specified task and model.zyolo predict model= source=! imgsz=32 save save_crop save_txtN)rrrr r r test_predict,sr$rcCstd|ddS)z2Test exporting a YOLO model to TorchScript format.zyolo export model=z format=torchscript imgsz=32Nr)rr r r test_export2sr%detectyolov8n-rtdetr.yaml coco8.yamlcCs^td|d|d|dtd|d|dtddtrZtd|d tddd S) zdTest the RTDETR functionality within Ultralytics for detection tasks using specified model and data.rrrz3 --imgsz= 160 epochs =1, cache = disk fraction=0.25z yolo predict r"bus.jpgz" imgsz=160 save save_crop save_txtz model='rtdetr-l.pt' source=N)rrr rr r r test_rtdetr8s r*z3MobileSAM with CLIP is not supported in Python 3.12)reasonsegmentz FastSAM-s.ptzcoco8-seg.yamlc Cstd}td|d|d|dtd|d|dd d lm}d d lm}||}|t|fD]P}||d d dddd}|j|d j j dd\} } ||gdddggdgddqhdS)z]Test FastSAM model for segmenting objects in images using various prompts within Ultralytics.r)zyolo segment val rrz imgsz=32zyolo segment predict model=r"r#r)FastSAM) PredictorcpuTi@g?g?)ZdeviceZ retina_masksZimgszconfZiou)Zmin_areaiii iza photo of a dog)bboxespointslabelsZtextsN) rr ultralyticsr-Zultralytics.models.samr.ropenZremove_small_regionsmasksr) rrrsourcer-r.Z sam_modelsZeverything_resultsZ new_masks_r r r test_fastsamAs  r>cCsNddlm}|td}td}|j|ddgdgd|j|gd d d d S) zATest MobileSAM segmentation with point prompts using Ultralytics.r)SAMz mobile_sam.ptz zidane.jpgiirr4)r6r7r2T)r5saveN)r8r?r rZpredict)r?rr;r r rtest_mobilesamZs   rAzCUDA is not availablezDDP is not availablecCs<td|d|d|dtd|d|d|ddS)z:Test YOLO training on GPU(s) for various tasks and models.rrrz imgsz=32 epochs=1 device=0z imgsz=32 epochs=1 device=0,1Nrrr r rtest_train_gpuosrC)r&r'r()"rZpytestZPILrtestsrrZultralytics.cfgrrrZultralytics.utilsrr r Zultralytics.utils.torch_utilsr ZTASK_MODEL_DATAZMODELSrrmarkZ parametrizer r!r$r%r*ZskipifZIS_PYTHON_3_12r>rAZslowrCr r r rs6