From 2e40b7f1138d523d128b6e3d939b63c85fec3139 Mon Sep 17 00:00:00 2001 From: alessandro faria Date: Mon, 5 Sep 2022 18:22:01 -0300 Subject: [PATCH] ADD weights yolov4 in models.yml --- ...olov3.txt => object_detection_classes_yolov4.txt} | 12 ++++++------ samples/dnn/models.yml | 10 +++++----- 2 files changed, 11 insertions(+), 11 deletions(-) rename samples/data/dnn/{object_detection_classes_yolov3.txt => object_detection_classes_yolov4.txt} (90%) diff --git a/samples/data/dnn/object_detection_classes_yolov3.txt b/samples/data/dnn/object_detection_classes_yolov4.txt similarity index 90% rename from samples/data/dnn/object_detection_classes_yolov3.txt rename to samples/data/dnn/object_detection_classes_yolov4.txt index 941cb4e139..ca76c80b5b 100644 --- a/samples/data/dnn/object_detection_classes_yolov3.txt +++ b/samples/data/dnn/object_detection_classes_yolov4.txt @@ -1,8 +1,8 @@ person bicycle car -motorcycle -airplane +motorbike +aeroplane bus train truck @@ -55,12 +55,12 @@ pizza donut cake chair -couch -potted plant +sofa +pottedplant bed -dining table +diningtable toilet -tv +tvmonitor laptop mouse remote diff --git a/samples/dnn/models.yml b/samples/dnn/models.yml index 86e86925df..f6e9f0e41f 100644 --- a/samples/dnn/models.yml +++ b/samples/dnn/models.yml @@ -23,16 +23,16 @@ opencv_fd: # Might be used for all YOLOv2, TinyYolov2, YOLOv3, YOLOv4 and TinyYolov4 yolo: load_info: - url: "https://pjreddie.com/media/files/yolov3.weights" - sha1: "520878f12e97cf820529daea502acca380f1cb8e" - model: "yolov3.weights" - config: "yolov3.cfg" + url: "https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.weights" + sha1: "0143deb6c46fcc7f74dd35bf3c14edc3784e99ee" + model: "yolov4.weights" + config: "yolov4.cfg" mean: [0, 0, 0] scale: 0.00392 width: 416 height: 416 rgb: true - classes: "object_detection_classes_yolov3.txt" + classes: "object_detection_classes_yolov4.txt" sample: "object_detection" tiny-yolo-voc: