如何使用暗网检测视频中的物体?

Question

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

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

Answer 1

如果您仔细查看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后尝试上面的代码。