使用 python 进行 Yolo-v3 对象检测

Question

我正在尝试使用 Yolo-v3 检测对象，参考 this 教程。我已经安装了暗网。

当我尝试运行以下代码时：

from darknet import Darknet

上面写着：

ImportError: 无法从“darknet”（未知位置）导入名称“Darknet”

所以我尝试了：

import darknet as dn
net = dn.load_net("cfg/tiny-yolo.cfg", "tiny-yolo.weights", 0)

然后它说：

"AttributeError: 模块 'darknet' 没有属性 'load_net'"

如何避免这些错误？

Answer 1

像这样使用它。

import cv2
import matplotlib.pyplot as plt
from utils import *
from darknet import Darknet
net = Darknet("cfg/tiny-yolo.cfg")
net.load_weights("tiny-yolo.weights")