0001-opencv4.4调用darknet-yoloV4模型

怼烎@ 2023-01-07 07:09 262阅读 0赞

# 个人公众号 #

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3pob25nbG9uZ3NoZW4_size_16_color_FFFFFF_t_70_pic_center]

### 参考链接 ###

[opencv调用yoloV4][opencv_yoloV4]

import cv2 as cv
    import time
    
    net = cv.dnn_DetectionModel('hand_v4-tiny.cfg', 'hand_v4-tiny_30000.weights')
    net.setInputSize(416, 416)
    net.setInputScale(1.0 / 255)
    net.setInputSwapRB(True)
    
    frame = cv.imread('imgs/69.jpg')
    
    with open('hand.names', 'rt') as f:
      names = f.read().rstrip('\n').split('\n')
    print(names)
    startTime = time.time()
    classes, confidences, boxes = net.detect(frame, confThreshold=0.2, nmsThreshold=0.25)
    endTime = time.time()
    print("Time: {}s".format(endTime-startTime))
    for classId, confidence, box in zip(classes.flatten(), confidences.flatten(), boxes):
      label = '%.2f' % confidence
      label = '%s: %s' % (names[classId], label)
      labelSize, baseLine = cv.getTextSize(label, cv.FONT_HERSHEY_SIMPLEX, 0.5, 1)
      left, top, width, height = box
      top = max(top, labelSize[1])
      cv.rectangle(frame, box, color=(0, 255, 0), thickness=3)
      cv.rectangle(frame, (left, top - labelSize[1]), (left + labelSize[0], top + baseLine), (255, 255, 255), cv.FILLED)
      cv.putText(frame, label, (left, top), cv.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 0))
    
    cv.imshow('out', frame)
    cv.waitKey(0)

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3pob25nbG9uZ3NoZW4_size_16_color_FFFFFF_t_70_pic_center]: /images/20221119/25f9754c245d4e4d89a6a730e0578a3e.png
[opencv_yoloV4]: https://blog.csdn.net/weixin_44613063/article/details/108038602?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522161725702816780274140733%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=161725702816780274140733&biz_id=0