Добрый день! Я писал код по ролику (ссылка на ролик, если нужно
https://www.youtube.com/watch?v=Uj4O2_dwRiA), но при запуске у меня возникает ошибка
Traceback (most recent call last):
File "C:\Users\HP\Desktop\python\__pycache__\parsing\musik\test.py", line 1, in <module>
from imageai.Detection import ObjectDetection
File "C:\Users\HP\AppData\Roaming\Python\Python39\site-packages\imageai\Detection\__init__.py", line 17, in <module>
from imageai.Detection.YOLOv3.models import yolo_main, tiny_yolo_main
File "C:\Users\HP\AppData\Roaming\Python\Python39\site-packages\imageai\Detection\YOLOv3\models.py", line 8, in <module>
from keras.layers.normalization import BatchNormalization
ImportError: cannot import name 'BatchNormalization' from 'keras.layers.normalization' (C:\Users\HP\AppData\Roaming\Python\Python39\site-packages\keras\layers\normalization\__init__.py)
Что-то связанное с модулями, но я так и не понял. Прошу вашей помощи.
Сам код:
from imageai.Detection import ObjectDetection
import os
exect_path=os.getcwd()
detector = ObjectDetection()
detector.setModelTypeAsRetinaNet()
detector.setModelPath(os.path.join(exect_path, "resnet50_coco_best_v2.1.0.h5"))
detector.loadModel()
list_ = detector.detectObjectsFromImage(
input_image=os.path.join(exect_path, "scale_1200.jpg"),
output_image_path=os.path.join(exect_path, "img.png")
)
