o Ďi8@sdZddlZddlZddlZddlZddlmZddlZee Z e j dZ e e ejvr6eje e eeje eZ ddlmZddlmZmZmZmZmZddlmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(ddl)m*Z*m+Z+m,Z,ddl-m.Z.m/Z/e/d e d e d ddd d dddddddddddddddddddddfddZ0ddZ1ddZ2e3dkre2dSdS)a Run YOLOv5 detection inference on images, videos, directories, globs, YouTube, webcam, streams, etc. Usage - sources: $ yolov5 detect --weights yolov5s.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/Zgi9g1ksQHc' # YouTube 'rtsp://example.com/media.mp4' # RTSP, RTMP, HTTP stream Usage - formats: $ yolov5 detect --weights yolov5s.pt # PyTorch yolov5s.torchscript # TorchScript yolov5s.onnx # ONNX Runtime or OpenCV DNN with --dnn yolov5s_openvino_model # OpenVINO yolov5s.engine # TensorRT yolov5s.mlmodel # CoreML (macOS-only) yolov5s_saved_model # TensorFlow SavedModel yolov5s.pb # TensorFlow GraphDef yolov5s.tflite # TensorFlow Lite yolov5s_edgetpu.tflite # TensorFlow Edge TPU yolov5s_paddle_model # PaddlePaddle N)Path)DetectMultiBackend) IMG_FORMATS VID_FORMATS LoadImagesLoadScreenshots LoadStreams)LOGGERProfile check_filecheck_img_size check_imshowcheck_requirementscolorstrcv2increment_pathnon_max_suppression print_args scale_boxesstrip_optimizer xyxy2xywh) Annotatorcolors save_one_box) select_devicesmart_inference_mode yolov5s.pt data/imagesdata/coco128.yaml??F runs/detectexpcI st|}| o |d }t|jddttv}|d}|p.|dp.|o.| }|d} |r>|r>t |}|durI|durId}n|durO|}t |t rX||g}t t|||d}!| rh|!dn|!j d d d t|}t|||||d }"|"j|"j|"j}#}$}%t||#d }d}&|rtd d } t|||#|%|d}'t|'}&n| rt|||#|%d}'n t|||#|%|d}'dg|&dg|&}(})|"j|%s|"jrdn|&dg|Rddgtttf}*}+|'D]\},}-}.}/}0|+d-t|-|"j}-|"j r|-!n|-"}-|-d}-t|-j#dkr|-d}-Wdn 1s"wY|+d|r:t |!t|,j$d dnd}|"|-||d}1Wdn 1sNwY|+dt%|1|||||d}1Wdn 1smwYt&|1D]\}2}3d7|r|,|2|.|2'|'j(}4}5}6|0|2d7}0n|,|.'t)|'dd}4}5}6t|4}4t|!|4j*}7t|!d|4j$|'j+dkrdnd|6}8|0d|-j#dd7}0t,|5j#gd}9| r|5'n|5}:t-|5|t|$d };t|3rt.|-j#dd|3dddd!f|5j#/|3dddd!f<|3ddd"f0D]#}<|3ddd"f|}?}@| rt3t,|>4dd!|94d&5}A| rj|@g|A|?Rn|@g|AR}Bt6|8dd'}C|C7d(t|B8|Bd)Wdn 1swY|s| s| rt |@}<|rdn|r|$|<n |$|<d#|?d*}D|;j9|>|Dt:||:|!d,|$|<|4j$d-d d.qD|;<}5| rt=>d/kr|4|*vr|*?|4t@At|4t@jBt@jCBt@Dt|4|5j#d|5j#dt@Et|4|5t@Fd|r|'j+dkr0t@G|7|5qv|(|2|7kr|7|(|2<t |)|2t@jHrJ|)|2I|/rd|/Jt@jK}Et |/Jt@jL}Ft |/Jt@jM}Gnd0|5j#d|5j#d}E}F}Gtt|7Nd1}7t@H|7t@jOd2|E|F|Gf|)|2<|)|27|5qvtPQ|0t|3rdnd3|+djRd4d5d6qtSfd7d8|+D}HtPQd9ddg|R|H| s|r| rd)ttT|!Ud:d;|!dnd}0tPQdNz.txtr&)zrtsp://zrtmp://zhttp://zhttps://z.streamsscreen)exist_oklabelsT)parentsr))devicednndatafp16)s)warn)img_sizestrideauto vid_stride)r2r3r4r%)imgszr)mkdirF)augment visualize)max_detz: frameimager"_z%gx%g )r&rr&r) line_widthexample r0z, az%g  z.2f)colorcropsz.jpg)fileBGRLinuxz.mp4mp4vz(no detections), @@z.1fmsc3s|] }|jdVqdS)rON)t).0xseenI/home/jeff/fluffinator/venv/lib/python3.10/site-packages/yolov5/detect.py szrun..zKSpeed: %.1fms pre-process, %.1fms inference, %.1fms NMS per image at shape z labels/*.txtz labels saved to zResults saved to bold)Xstrendswithrsuffixrrlower startswith isnumericr isinstanceintrr8rrr3namesptr r rlenrrwarmuptritonr torch from_numpytor,r/halffloatshapestemr enumeratecopycountgetattrnamemodetensorrrrounduniquesumreversedrviewtolistopenwriterstrip box_labelrrresultplatformsystemappendr namedWindow WINDOW_NORMALWINDOW_KEEPRATIO resizeWindowimshowwaitKeyimwrite VideoWriterreleaseget CAP_PROP_FPSCAP_PROP_FRAME_WIDTHCAP_PROP_FRAME_HEIGHT with_suffixVideoWriter_fourccr infodttuplelistglobrr)Iweightssourcer.r6img conf_thres iou_thresr<r,view_imgsave_txt save_conf save_cropnosaveclasses agnostic_nmsr9r:updateprojectrrr)line_thickness hide_labels hide_confrjr-r5save_imgis_fileis_urlwebcam screenshotsave_dirmodelr3rbrcbsdatasetvid_path vid_writerwindowsrpathimim0svid_capr0predidetpim0r= save_pathtxt_pathgnimc annotatorcnxyxyconfclsxywhlineflabelfpswhrQrVrTrWrun5s    $    , <,$" ,(     4 ,rc Cst}|jddtddd|jdttddd |jd ttd d d |jd dddtdgdd|jdtddd |jdtddd |jdtddd |jdddd|jdd d!d"|jd#d d$d"|jd%d d&d"|jd'd d(d"|jd)d d*d"|jd+dtd,d-|jd.d d/d"|jd0d d1d"|jd2d d3d"|jd4d d5d"|jd6d7d8d|jd9d:d8d|jd;d dtd?d@|jdAdBd dCdD|jdEdBd dFdD|jdGd dHd"|jdId dJd"|jdKtdLdMd |}|jt |jdLkrdNndL9_t t ||S)ONz --weights+rzmodel path or triton URL)nargstypedefaulthelpz--sourcerz"file/dir/URL/glob/screen/0(webcam))rrrz--datarz(optional) dataset.yaml pathz--imgszz--imgz --img-sizer(zinference size h,wz --conf-thresrzconfidence thresholdz --iou-thresr zNMS IoU thresholdz --max-detr!zmaximum detections per imagez--devicer"z%cuda device, i.e. 0 or 0,1,2,3 or cpu)rrz --view-img store_truez show results)actionrz --save-txtzsave results to *.txtz --save-confz%save confidences in --save-txt labelsz --save-cropzsave cropped prediction boxesz--nosavezdo not save images/videosz --classesz0filter by class: --classes 0, or --classes 0 2 3)rrrz--agnostic-nmszclass-agnostic NMSz --augmentzaugmented inferencez --visualizezvisualize featuresz--updatezupdate all modelsz --projectr#zsave results to project/namez--namer$z --exist-okz*existing project/name ok, do not incrementz--line-thicknessr%zbounding box thickness (pixels))rrrz --hide-labelsFz hide labels)rrrz --hide-confzhide confidencesz--halfz!use FP16 half-precision inferencez--dnnz!use OpenCV DNN for ONNX inferencez --vid-strider&zvideo frame-rate strider;) argparseArgumentParser add_argumentrZROOTrark parse_argsr6rdrvars)parseroptrVrVrW parse_opts@" rcCs&t}tddtdit|dS)N) tensorboardthop)excluderV)rrrr)rrVrVrWmains r__main__)4__doc__rosrsyspathlibrrg__file__resolveFILEr+rrZrrrelpathcwdyolov5.models.commonryolov5.utils.dataloadersrrrrryolov5.utils.generalr r r r r rrrrrrrrryolov5.utils.plotsrrryolov5.utils.torch_utilsrrrrr__name__rVrVrVrWsj    @ /#