Question

Привет! Я делаю распознавание объектов в python. Вот мой код:
from imageai.Detection import ObjectDetection
detector = ObjectDetection()
model_path = “./models/yolo-tiny.h5”
input_path = “./input/test45.jpg”
output_path = “./output/newimage.jpg”
detector.setModelTypeAsTinyYOLOv3()
detector.setModelPath(model_path)
detector.loadModel()
detection = detector.detectObjectsFromImage(input_image=input_path, output_image_path=output_path)
for eachItem in detection:
print(eachItem , “ : ”, eachItem)
Этот код выдает ошибку:
Traceback (most recent call last):
File “C:\Users\Даниил\PycharmProjects\object_detection\main.py”, line 1, in
from imageai.Detection import ObjectDetection
File “C:\Users\Даниил\PycharmProjects\object_detection\venv\lib\site-packages\imageai\Detection\__init__.py”, line 17, in
from imageai.Detection.YOLOv3.models import yolo_main, tiny_yolo_main
File “C:\Users\Даниил\PycharmProjects\object_detection\venv\lib\site-packages\imageai\Detection\YOLOv3\models.py”, line 8, in
from keras.layers.normalization import BatchNormalization
ImportError: cannot import name ‘BatchNormalization’ from ‘keras.layers.normalization’ (C:\Users\Даниил\PycharmProjects\object_detection\venv\lib\site-packages\keras\layers\normalization\__init__.py)
Помогите решить проблему. Заранее спасибо.

Answer 1

Такие проблемы на данном сайте не решить. У тебя ошибка импорта пакетов, почему - отсюда не видно.

Почти уверен, что напутано с зависимостями.

Под каждый подобный проект создавай отдельные среды (venv, conda env, и тд), и устанавливай пакеты по новой.

Если все сделано заново и точно по гайдам, и все равно не работает - посмотри, когда обновляли imageai, и конфликтующие библиотеки. Может быть там что-то обновили, и оно теперь не хочет работать. Ну, или можно попробовать установить пакеты более ранней версии

Вообще тензорфлоу и керас - не самые приятные либы, керас - так вообще боль. Если сильно хочется попробовать детекцию через йоло - найди ultralytics/yolov5 на гитхабе.