使用Python的Yolo-v3对象检测

Question

我正在尝试使用this教程的Yolo-v3检测对象。我已经安装了darknet。

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

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")