当前位置:   article > 正文

python调用YOLOv4训练的.weights模型_weights是什么模型

weights是什么模型

调用代码(网上很多可以参考)

import numpy as np
import cv2

def video_demo():
    # 加载已经训练好的模型路径,可以是绝对路径或者相对路径
    weightsPath = "backup/yolo-obj_final.weights"
    configPath = "cfg/yolo-obj.cfg"
    labelsPath = "data/coco.names"
    # 初始化一些参数
    LABELS = open(labelsPath).read().strip().split("\n")  # 物体类别
    COLORS = np.random.randint(0, 255, size=(len(LABELS), 3), dtype="uint8")  # 颜色
    net = cv2.dnn.readNetFromDarknet(configPath, weightsPath)
    net.setPreferableBackend(cv2.dnn.DNN_BACKEND_CUDA)
    net.setPreferableTarget(cv2.dnn.DNN_TARGET_CUDA)

    # 读入待检测的图像
    # 0是代表摄像头编号,只有一个的话默认为0
    capture = cv2.VideoCapture(0)
    # 读入待检测的图像
    # 0是代表摄像头编号,只有一个的话默认为0
    yolo_num = 1
    while (True):
        boxes = []
        confidences = []
        classIDs = []
        print("yolo%d" % yolo_num)
        yolo_num = yolo_num
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/weixin_40725706/article/detail/128665
推荐阅读
相关标签
  

闽ICP备14008679号