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

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

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

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"

Как я могу избежать этих ошибок?


person Totura    schedule 15.10.2019    source источник


Ответы (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")
person AbdelAziz AbdelLatef    schedule 15.10.2019
comment
Большое тебе спасибо. Вместо исходного darknet.py я использовал этот github .com / Garima13a / YOLO-Object-Detection / blob / master /. - person Totura; 15.10.2019
comment
Я не могу найти это dn.load_net по этой ссылке. Попробуй мой ответ сейчас. - person AbdelAziz AbdelLatef; 15.10.2019