a yf2;@sddlZddlmZddlZddlZddlmZmZm Z ddl m Z ddl m Z mZddlmZddlmZmZmZddlmZmZGd d d e ZdS) N)Path)build_dataloaderbuild_yolo_dataset converter) BaseValidator)LOGGERops)check_requirements)ConfusionMatrix DetMetricsbox_iou)output_to_target plot_imagescseZdZdZd*fdd ZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZd+ddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)ZZS),DetectionValidatoraR A class extending the BaseValidator class for validation based on a detection model. Example: ```python from ultralytics.models.yolo.detect import DetectionValidator args = dict(model="yolov8n.pt", data="coco8.yaml") validator = DetectionValidator(args=args) validator() ``` Ncst|||||d|_d|_d|_d|_d|_d|j_t |j |j d|_ t ddd|_|j|_g|_|jjrtddS) zAInitialize detection model with necessary variables and settings.NFdetect)save_diron_plotg?gffffff? uWARNING ⚠️ 'save_hybrid=True' will append ground truth to predictions for autolabelling. WARNING ⚠️ 'save_hybrid=True' will cause incorrect mAP. )super__init__ nt_per_class nt_per_imageis_cocois_lvis class_mapargstaskr rrmetricstorchZlinspaceZiouvZnumelnioulb save_hybridrwarning)self dataloaderrZpbarr _callbacks __class__^/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/models/yolo/detect/val.pyrs zDetectionValidator.__init__csdj|jddd<|jjr,dn ddd<dD]}||j|<qD|jjrˆdjdd\}}td}dtj ||||f|jd fd d t |D|_ S) z/Preprocesses batch of images for YOLO training.imgT)Z non_blocking) batch_idxclsbboxesNr.devicecs:g|]2}tjdd|kd|kgddqS)r-r,)dim)rcat).0ibatchr.r(r) =sz1DetectionValidator.preprocess..) tor1rZhalffloatr!shapelenrtensorranger )r#r8kheightwidthnbr(r7r) preprocess2s(   zDetectionValidator.preprocesscCs |j|jjd}t|toFd|voF|tjdpF|tjd|_ t|tobd|vob|j |_ |j rtt nt tt|j|_|jj|j s|j o|j O_|j|_t|j|_|j|j_|jj|j_t|j|jjd|_d|_g|_tgggggd|_d S) z'Initialize evaluation metrics for YOLO.Zcocoz val2017.txtztest-dev2017.txtlvis)ncconfr)tprHpred_cls target_cls target_imgN)datagetrsplit isinstancestrendswithosseprrrZcoco80_to_coco91_classlistr?r=namesr save_jsontrainingrGrplotsplotr rHconfusion_matrixseenjdictdictstats)r#modelvalr(r(r) init_metricsDs" ""    zDetectionValidator.init_metricscCsddS)zBReturn a formatted string summarizing class metrics of YOLO model.z%22s%11s%11s%11s%11s%11s%11s)ClassZImagesZ InstanceszBox(PRZmAP50z mAP50-95)r()r#r(r(r)get_descXszDetectionValidator.get_descc Cs2tj||jj|jj|jd|jjp&|jj|jjdS)z4Apply Non-maximum suppression to prediction outputs.T)labelsZ multi_labelZagnosticmax_det) rZnon_max_suppressionrrHiour single_clsZ agnostic_nmsrg)r#predsr(r(r) postprocess\szDetectionValidator.postprocessc Cs|d|k}|d|d}|d|}|d|}|djdd}|d |}t|rt|tj||jd gd }tj||||d |||||d S):Prepares a batch of images and annotations for validation.r,r-r2r. ori_shaper*r/N ratio_padr0)rrorrn)r-bboxrmimgszrn) squeezer<r=rZ xywh2xyxyrr>r1 scale_boxes) r#sir8idxr-rqrmrrrnr(r(r)_prepare_batchhs    "z!DetectionValidator._prepare_batchcCs:|}tj|d|ddddf|d|dd|S)rlrrNrmrnrp)clonerrt)r#predpbatchprednr(r(r) _prepare_predus $z DetectionValidator._prepare_predc Cst|D]\}}|jd7_t|}ttjd|jdtjd|jdtj||jtj|jdd}| ||}| d| d}} t|} ||d<| |d <|dkr| r|j D]} |j | || q|jjr|jjd | |d q|jjrd|d d d f<|||} | d d d f|d<| d d d f|d<| rl|| | ||d<|jjrl|j| | ||j D]} |j | || qv|jjr|| |d||jjr|| |jj|d|jdt|d|jdqd S)zMetrics.rorr0)dtyper1)rHrJrIr-rqrKrLN) detections gt_bboxesgt_clsrxrHrJrIim_filermrfz.txt) enumerater\r=r^rzerosr1rboolrwpopuniquer_keysappendrrYr[Z process_batchrir}_process_batchrW pred_to_jsonsave_txt save_one_txt save_confrrstem) r#rjr8rurzZnprstatr{r-rqnlr@r|r(r(r)update_metrics}sP       z!DetectionValidator.update_metricscOs|j|j_|j|j_dS)z8Set final values for metrics speed and confusion matrix.N)speedrr[)r#rkwargsr(r(r)finalize_metricss z#DetectionValidator.finalize_metricscCsdd|jD}tj|dt|jd|_tj|dt|jd|_| ddt |r~|d r~|j j fi||j jS)z2Returns metrics statistics and results dictionary.cSs&i|]\}}|t|dqS)r)rr4cpunumpy)r5r@vr(r(r) z0DetectionValidator.get_stats..rK)Z minlengthrLNrI)r_itemsnpZbincountZastypeintrGrrrr=anyrprocessZ results_dict)r#r_r(r(r) get_statss zDetectionValidator.get_statscCsddt|jj}t|d|j|jg|jR|jdkr`t d|j j d|j j r|j s|jdkrt|jrt|jjD]<\}}t||j||j||j|g|j|Rq|j jrdD]"}|jj|j|j||jd qd S) z1Prints training/validation set metrics per class.z %22s%11i%11iz%11.3gallru"WARNING ⚠️ no labels found in z, set, can not compute metrics without labelsro)TF)rrV normalizerN)r=rrrinfor\rsumZ mean_resultsr"rrverboserXrGr_rZap_class_indexrVrZ class_resultrYr[rZrvaluesr)r#pfr6crr(r(r) print_resultss(",z DetectionValidator.print_resultscCs4t||ddddf}||dddf||S)a> Return correct prediction matrix. Args: detections (torch.Tensor): Tensor of shape (N, 6) representing detections where each detection is (x1, y1, x2, y2, conf, class). gt_bboxes (torch.Tensor): Tensor of shape (M, 4) representing ground-truth bounding box coordinates. Each bounding box is of the format: (x1, y1, x2, y2). gt_cls (torch.Tensor): Tensor of shape (M,) representing target class indices. Returns: (torch.Tensor): Correct prediction matrix of shape (N, 10) for 10 IoU levels. Note: The function does not return any value directly usable for metrics calculation. Instead, it provides an intermediate representation used for evaluating predictions against ground truth. Nrxr)r Zmatch_predictions)r#rrrrhr(r(r)rsz!DetectionValidator._process_batchracCst|j|||j||jdS)aG Build YOLO Dataset. Args: img_path (str): Path to the folder containing images. mode (str): `train` mode or `val` mode, users are able to customize different augmentations for each mode. batch (int, optional): Size of batches, this is for `rect`. Defaults to None. )modestride)rrrMr)r#Zimg_pathrr8r(r(r) build_datasets z DetectionValidator.build_datasetcCs&|j||dd}t|||jjdddS)z Construct and return dataloader.ra)r8rFr2)shuffleZrank)rrrworkers)r#Z dataset_pathZ batch_sizedatasetr(r(r)get_dataloadersz!DetectionValidator.get_dataloaderc CsHt|d|d|dd|d|d|jd|d|j|jd d S) zPlot validation image samples.r*r,r-r2r.r val_batchz _labels.jpgpathsfnamerVrN)rrsrrVr)r#r8nir(r(r)plot_val_sampless z#DetectionValidator.plot_val_samplescCsFt|dgt||jjdR|d|jd|d|j|jddS)zDPlots predicted bounding boxes on input images and saves the result.r*)rgrrz _pred.jpgrN)rr rrgrrVr)r#r8rjrr(r(r)plot_predictionssz#DetectionValidator.plot_predictionscCsTddlm}|tj|d|dftjdd|j|ddddfdj||ddS) zRSave YOLO detections to a txt file in normalized coordinates in a specific format.r)Resultsro)r~N)pathrVZboxes)r)Zultralytics.engine.resultsrrrZuint8rVr)r#r|rr<filerr(r(r)rs zDetectionValidator.save_one_txtc Cst|j}|rt|n|}t|ddddf}|ddddf|ddddfd8<t||D]N\}}|j ||j t|d|j rdnddd|Dt |ddd qzdS) z/Serialize YOLO predictions to COCO json format.Nrxr/rrorcSsg|]}t|dqS))roundr5xr(r(r)r9%rz3DetectionValidator.pred_to_json..)image_idZ category_idrqZscore) rr isnumericrrZ xyxy2xywhziptolistr]rrrr)r#r|filenamerrboxpbr(r(r)rs 0   zDetectionValidator.pred_to_jsonc Cs|jjr|js|jrt|jr|jd}|jdd|jrDdnd|jjd}|jr`dnd}t d |d |d |d z>||fD]}| sJ|d qt |jrdnd|jrddl m}ddlm}|t|}|t|} ||| d} n6ddlm} m} | t|}|t|} | || d} dd|jjjD| j_| | | |jrz| |jr| j ddn| j!d| j!dg\||j"j#d<||j"j#d<Wn:t$y} z t %|d| WYd} ~ n d} ~ 00|S)zHEvaluates YOLO output in JSON format and returns performance statistics.zpredictions.jsonr annotationszinstances_val2017.jsonZlvis_v1_z.jsonZ pycocotoolsrFz Evaluating z mAP using z and z...z file not foundzpycocotools>=2.0.6z lvis>=0.5.3r)COCO)COCOevalrq)LVISLVISEvalcSsg|]}tt|jqSr()rrrrr(r(r)r9Frz0DetectionValidator.eval_json..Nr/ZAP50ZAPr2z unable to run: )&rrWrrr=r]rrMrOrris_filer Zpycocotools.cocorZpycocotools.cocoevalrrQZloadResrFrrZ _load_jsonr$rZim_filesparamsZimgIdsevaluate accumulateZ summarizerr_resultsrr Exceptionr")r#r_Z pred_jsonZ anno_jsonpkgrrrannorzrarrer(r(r) eval_json*sF$       ("*zDetectionValidator.eval_json)NNNNN)raN)__name__ __module__ __qualname____doc__rrDrbrerkrwr}rrrrrrrrrrrr __classcell__r(r(r&r)rs(   1     r)rSpathlibrrrrZultralytics.datarrrZultralytics.engine.validatorrZultralytics.utilsrrZultralytics.utils.checksr Zultralytics.utils.metricsr r r Zultralytics.utils.plottingr rrr(r(r(r)s