a yf?D@sdZddlZddlZddlZddlmZddlZddlZddl Z ddl m Z m Z ddl mZddlmZmZddlmZddlmZmZmZmZmZmZmZdd lmZmZdd lm Z dd l!m"Z"m#Z#d Z$Gd ddZ%dS)a Run prediction on images, videos, directories, globs, YouTube, webcam, streams, etc. Usage - sources: $ yolo mode=predict model=yolov8n.pt source=0 # webcam img.jpg # image vid.mp4 # video screen # screenshot path/ # directory list.txt # list of images list.streams # list of streams 'path/*.jpg' # glob 'https://youtu.be/LNwODJXcvt4' # YouTube 'rtsp://example.com/media.mp4' # RTSP, RTMP, HTTP, TCP stream Usage - formats: $ yolo mode=predict model=yolov8n.pt # PyTorch yolov8n.torchscript # TorchScript yolov8n.onnx # ONNX Runtime or OpenCV DNN with dnn=True yolov8n_openvino_model # OpenVINO yolov8n.engine # TensorRT yolov8n.mlpackage # CoreML (macOS-only) yolov8n_saved_model # TensorFlow SavedModel yolov8n.pb # TensorFlow GraphDef yolov8n.tflite # TensorFlow Lite yolov8n_edgetpu.tflite # TensorFlow Edge TPU yolov8n_paddle_model # PaddlePaddle yolov8n_ncnn_model # NCNN N)Path)get_cfg get_save_dir)load_inference_source) LetterBoxclassify_transforms) AutoBackend) DEFAULT_CFGLOGGERMACOSWINDOWS callbackscolorstrops) check_imgsz check_imshow)increment_path) select_devicesmart_inference_modeu" WARNING ⚠️ inference results will accumulate in RAM unless `stream=True` is passed, causing potential out-of-memory errors for large sources or long-running streams and videos. See https://docs.ultralytics.com/modes/predict/ for help. Example: results = model(source=..., stream=True) # generator of Results objects for r in results: boxes = r.boxes # Boxes object for bbox outputs masks = r.masks # Masks object for segment masks outputs probs = r.probs # Class probabilities for classification outputs c@seZdZdZeddfddZddZddZd d Zd d Z d&ddZ d'ddZ ddZ e d(ddZd)ddZddZd*ddZd+dd Zed!d"d#Zed!d$d%ZdS), BasePredictoraI BasePredictor. A base class for creating predictors. Attributes: args (SimpleNamespace): Configuration for the predictor. save_dir (Path): Directory to save results. done_warmup (bool): Whether the predictor has finished setup. model (nn.Module): Model used for prediction. data (dict): Data configuration. device (torch.device): Device used for prediction. dataset (Dataset): Dataset used for prediction. vid_writer (dict): Dictionary of {save_path: video_writer, ...} writer for saving video output. NcCst|||_t|j|_|jjdur,d|j_d|_|jjrHtdd|j_d|_|jj |_ d|_ d|_ d|_ i|_ d|_d|_d|_g|_d|_d|_d|_|pt|_d|_t|_t|dS)z Initializes the BasePredictor class. Args: cfg (str, optional): Path to a configuration file. Defaults to DEFAULT_CFG. overrides (dict, optional): Configuration overrides. Defaults to None. Ng?FT)warnr)rargsrsave_dirconf done_warmupshowrmodeldataimgszdevicedataset vid_writer plotted_img source_typeseenwindowsbatchresults transformsr Zget_default_callbackstxt_path threadingLock_lockZadd_integration_callbacks)selfcfgZ overrides _callbacksr0X/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/engine/predictor.py__init__Ps0     zBasePredictor.__init__cCst|tj }|rNt||}|ddddfd}t|}t|}| |j }|j j rj| n|}|r~|d}|S)z Prepares input image before inference. Args: im (torch.Tensor | List(np.ndarray)): BCHW for tensor, [(HWC) x B] for list. .N)r) isinstancetorchTensornpstack pre_transformZ transposeZascontiguousarrayZ from_numpytorrfp16halffloat)r-imZ not_tensorr0r0r1 preprocessss   zBasePredictor.preprocesscOs^|jjr2|jjs2t|jt|jddjddnd}|j |g|R|jj ||jj d|S)zHRuns inference on a given image using the specified model and arguments.rT)mkdirF)augment visualizeembed) rrFr#tensorrrrr&stemrrErG)r-rBrkwargsrFr0r0r1 inferences$zBasePredictor.inferencecsDtdd|Ddk}t|j|o&|jj|jjdfdd|DS)z Pre-transform input image before inference. Args: im (List(np.ndarray)): (N, 3, h, w) for tensor, [(h, w, 3) x N] for list. Returns: (list): A list of transformed images. cSsh|] }|jqSr0)shape.0xr0r0r1 z.BasePredictor.pre_transform..r5)autostridecsg|]}|dqS))imager0rMZ letterboxr0r1 rQz/BasePredictor.pre_transform..)lenrrrptrS)r-rBZ same_shapesr0rUr1r=s zBasePredictor.pre_transformcCs|S)z9Post-processes predictions for an image and returns them.r0)r-predsZimgZ orig_imgsr0r0r1 postprocessszBasePredictor.postprocessFcOsF||_|r$|j||g|Ri|St|j||g|Ri|SdS)z)Performs inference on an image or stream.N)streamstream_inferencelist)r-sourcerr[rrJr0r0r1__call__szBasePredictor.__call__cCs|||}|D]}qdS)aK Method used for Command Line Interface (CLI) prediction. This function is designed to run predictions using the CLI. It sets up the source and model, then processes the inputs in a streaming manner. This method ensures that no outputs accumulate in memory by consuming the generator without storing results. Note: Do not modify this function or remove the generator. The generator ensures that no outputs are accumulated in memory, which is critical for preventing memory issues during long-running predictions. N)r\)r-r^rgen_r0r0r1 predict_clis zBasePredictor.predict_clicCst|jj|jjdd|_|jjdkrFt|jjdt|jd|jjdnd|_ t ||jj |jj |jj d|_|jj|_t|d d s|jjs|jjst|jd kstt|jd d grtti|_dS)z"Sets up source and inference mode.r6)rSZmin_dimZclassifyr(r) crop_fractionN)r^r& vid_stridebufferr[TiZ video_flagF)rrrrrStaskgetattrrrcr(rr&rdZ stream_bufferr r#r[Z screenshotrWanyr warningSTREAM_WARNINGr!)r-r^r0r0r1 setup_sources6      zBasePredictor.setup_sourcec /s*jjrtdjs"|j|dur<|njjjj sVjj rxjj rhj dnj j dddj sjjjjsjjrdnjjdgjRdd_ d gd___tjjd tjjd tjjd f}d jD]_d j\}}}|d |} Wdn1sL0Y|d`j| g|Ri|} jjrt| tj r| gn| EdHWdqWdn1s0Y|d  !| | |_"Wdn1s0Ydt#|} t$| D]} jd7_|d j%d| |dj%d| |d j%d| dj"| _&jjsjj sjj sjj'r$|| (| t)|| | |7<q$jjrtd*|dj"EdHqWdn1s0Yj+,D]} t| t-j.r"| /q"jjrjrt0fdd|D}tdt1jjjdg| j2d dR|jj sjj sjj3rt#t4j 5d}jj rd|dd|dkdj dnd}tdt6dj |ddS)zEStreams real-time inference on camera feed and saves results to file.NlabelsTparentsexist_okr5r4)rr)rZon_predict_startZon_predict_batch_startr6Zon_predict_postprocess_end@@)rCrKrZ Zon_predict_batch_endc3s|]}|jjdVqdS)rqN)tr$rMr-r0r1 "rQz1BasePredictor.stream_inference..zRSpeed: %.1fms preprocess, %.1fms inference, %.1fms postprocess per image at shape z labels/*.txtz labelsz saved to zResults saved to boldZon_predict_end)7rverboser infor setup_modelr,rkr^savesave_txtrrDrZwarmuprXZtritonr bsrr$r%r&rZProfiler run_callbacksrCrKrGr8r9r:rZr'rWrangedtspeedr write_resultsrjoinr!valuescv2 VideoWriterreleasetupleminrL save_cropr]globr)r-r^rrrJZ profilerspathsZim0srvrBrYnivrsnlr0rtr1r\sx   "2       *  0 0  ((  0 "2zBasePredictor.stream_inferenceTc Csbt|p |jjt|jj|d|jj|jj|jj|jjd|d|_|jj|_|jj |j_|j dS)zJInitialize YOLO model with given parameters and set it to evaluation mode.)rxT)weightsrdnnrr?r&ZfuserxN) rrrrrrrr@r&r?eval)r-rrxr0r0r1rz-s   zBasePredictor.setup_modelc Csd}t|jdkr|d}|jjs2|jjs2|jjrJ||d7}|jj}n$t d||}|rjt |dnd}|j d|j |jj dkrdnd ||_|d j|jd d7}|j|}|j |_ |||jd d d7}|jjs|jjr0|j|jj|jj|jj|jj|jjr"dn||d|_|jjrT|j|jd|jjd|jj rv|j |j d|jj d|jjr|t!||jjr|"t!|j |j#||S)z/Write inference results to a file or directory.rlr4Nz: z frame (\d+)/r5rmrTraz {:g}x{:g} r6rKz.1fms) line_widthZboxesrrmZim_gpuz.txt) save_confZcrops)r file_name)$rWrLr#r[Zfrom_imgrHr countresearchintrrImoder)formatr'__str__rxrrr{rZplotrZ show_boxesZ show_confZ show_labelsZ retina_masksr"r|rrstrsave_predicted_imagesname) r-rprBrvstringframematchresultr0r0r1r>s< ,      zBasePredictor.write_resultsrlrcCs|j}|jjdvr|jjdkr&|jjnd}|dddd}||jvr|jjrdt|j ddd t rld n t rtd nd \}}t j tt||t j|||jd|jdfd |j|<|j|||jjrt ||d|n t ||dS)z0Save video predictions as mp4 at specified path.>r[videor.r5rz_frames/Trn)z.mp4Zavc1).aviZWMV2)rZMJPG)filenamefourccfpsZ frameSizez.jpgN)r"r rrsplitr!rZ save_framesrrDr r rrr with_suffixZVideoWriter_fourccrLwriteZimwrite)r-Z save_pathrrBrZ frames_pathsuffixrr0r0r1rfs$   z#BasePredictor.save_predicted_imagescCs|j}tdkrV||jvrV|j|t|tjtjBt ||j d|j dt ||t |j jdkrvdnddS)z>Display an image in a window using the OpenCV imshow function.Linuxr5rrTi,N)r"platformsystemr%appendrZ namedWindowZ WINDOW_NORMALZWINDOW_KEEPRATIOZ resizeWindowrLZimshowZwaitKeyr r)r-rrBr0r0r1rs  zBasePredictor.show)eventcCs |j|gD] }||qdS)z3Runs all registered callbacks for a specific event.N)r get)r-rcallbackr0r0r1r~szBasePredictor.run_callbackscCs|j||dS)z Add callback.N)r r)r-rfuncr0r0r1 add_callbackszBasePredictor.add_callback)NNF)NN)NN)T)rlr)rl)__name__ __module__ __qualname____doc__r r2rCrKr=rZr_rbrkrr\rzrrrrr~rr0r0r0r1r?s"#    V (  r)&rrrr*pathlibrrnumpyr;r9Zultralytics.cfgrrZultralytics.datarZultralytics.data.augmentrrZultralytics.nn.autobackendrZultralytics.utilsr r r r r rrZultralytics.utils.checksrrZultralytics.utils.filesrZultralytics.utils.torch_utilsrrrjrr0r0r0r1s"   $