164 lines (161 with data), 29.6 kB
§
¢eÜP ã óø d Z ddlZddlZddlZddlZddlZddlmZ ddlZ ddl
Z
ddlmZ ee¦ «
¦ « Zej d Z ee¦ « ej vr#ej ee¦ « ¦ « eej e ej ¦ « ¦ « ¦ « ZddlmZ ddlmZ ddlmZ ddlmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+ dd l,m-Z-m.Z.m/Z/ dd
l0m1Z1m2Z2m3Z3 ddl4m5Z5m6Z6 d Z7d
Z8d Z9 e6¦ « ddddddddddddddddedz dddddd ed¦ « d e¦ « dfd¦ « Z:d Z;d Z<e=dk r e;¦ « Z> e<e>¦ « dS dS )aç
Validate a trained YOLOv5 detection model on a detection dataset
Usage:
$ python val.py --weights yolov5s.pt --data coco128.yaml --img 640
Usage - formats:
$ python val.py --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)Útqdm)ÚDetectMultiBackend)Ú Callbacks)Úcreate_dataloader)ÚLOGGERÚTQDM_BAR_FORMATÚProfileÚ
check_datasetÚcheck_img_sizeÚcheck_requirementsÚ
check_yamlÚcoco80_to_coco91_classÚcolorstrÚincrement_pathÚnon_max_suppressionÚ
print_argsÚscale_boxesÚ xywh2xyxyÚ xyxy2xywh)ÚConfusionMatrixÚap_per_classÚbox_iou)Úoutput_to_targetÚplot_imagesÚplot_val_study)Ú
select_deviceÚsmart_inference_modec ó t j |¦ « g d¢ }| ¦ « D ]Ø^ }}}t t j |¦ « dd¦ « ¦ « |z d¦ « ¦ « }|r|g|¢|R n|g|¢R } t |d¦ « 5 }
|
dt | ¦ « z ¦ « | z dz ¦ « d d d ¦ « n# 1 swxY w Y Ùd S )N)é r r r r é éÿÿÿÿÚaz%g ú
) ÚtorchÚtensorÚtolistr ÚviewÚopenÚwriteÚlenÚrstrip)ÚprednÚ save_confÚshapeÚfileÚgnÚxyxyÚconfÚclsÚxywhÚlineÚfs ú2/home/iml/Desktop/Talha/YOLOV5_Model/yolov5/val.pyÚsave_one_txtr9 2 sW å eÑ Ô \\\Ô *BØ!LLNNð @ð @ÑtSÝ%, tÑ,Ô,×1Ò1°!°QÑ7Ô7Ñ8Ô8¸2Ñ=×CÒCÀBÑGÔG×NÒNÑPÔPØ%.Ð@Ð!dÐ!DÐ!Ð!Ð!°S°L¸4°L°LÝ
$__ð @ Ø
GGUS YYÑ&×.Ò.Ñ0Ô0°4Ñ7¸$Ñ>Ñ?Ô?Ð?ð @ð @ð @ñ @ô @ð @ð @ð @ð @ð @ð @øøøð @ð @ð @ð @øð@ð @s Â4>C>Ã>D ÄD c
ó |j ¦ « rt |j ¦ « n|j }t | d d
d d
f ¦ « }|d d
d d
fxx |d d
dd
f dz z cc<