Обнаружение объектов Yolo-v3 с помощью Python

Question

Я пытаюсь обнаружить объекты, используя Yolo-v3, ссылаясь на это руководство. Я уже установил даркнет.

Когда я пытался запустить следующий код:

from darknet import Darknet

, он говорит:

ImportError: невозможно импортировать имя 'Darknet' из 'darknet' (неизвестноlocation)

, поэтому я попытался:

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