时间戳

首页 > 解决方案 > 如何使用暗网检测视频中的对象？

python - 如何使用暗网检测视频中的对象？

问题描述

我正在使用暗网来检测图像中的对象，这非常有帮助。此外，我还想从计算机中的视频中检测对象。

我安装了 Open CV，但我的电脑没有 GPU。我认为我应该更改我的 darknet.py 文件中的某些内容。但是我应该添加什么代码？你能给我一个明确的指示吗？预先感谢。

标签： pythonpython-3.xyolo

解决方案

如果您仔细查看该darknet.py文件，您会发现一个关于如何从给定图像中提取对象的示例。你试过了吗？如果您不这样做，我建议您在进入视频之前先尝试一下。

然后，您将需要进入视频。这是一个文档链接，OpenCV它们解释了如何读取视频文件和流。从此代码中，您将能够提取视频的帧。

然后您可以使用其中的代码darkent.py并将帧提供给它。

如果没有，则有可能有用的深色包装纸。
这是一个链接。您可以阅读一份文档并将其安装在您的机器上。这是非常容易使用。它已经有一个关于如何在图像上使用它的示例。您可以修改它并使用视频。

import numpy as np
import cv2
import pyyolo

cap = cv2.VideoCapture('vtest.avi')
meta_filepath = "/home/rameshpr/Downloads/darknet_google_server/data/obj.data"
cfg_filepath = "/home/rameshpr/Downloads/darknet_google_server/cfg/yolo-lb.cfg"
weights_filepath = "/home/rameshpr/Downloads/darknet_google_server/backup/yolo-v3.weights"


meta = pyyolo.load_meta(meta_filepath)
net = pyyolo.load_net(cfg_filepath, weights_filepath, False)

while(cap.isOpened()):
    ret, frame = cap.read()
    if not ret:
        break

    yolo_img = pyyolo.array_to_image(frame)
    res = pyyolo.detect(net, meta, yolo_img)

    for r in res:
        cv2.rectangle(frame, r.bbox.get_point(pyyolo.BBox.Location.TOP_LEFT, is_int=True),
                      r.bbox.get_point(pyyolo.BBox.Location.BOTTOM_RIGHT, is_int=True), (0, 255, 0), 2)


    cv2.imshow('frame', frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

安装完成后试试上面的代码pyyolo。

推荐阅读