0

我尝试安装 pytorch 但失败了。我还下载了它运行所需的所有库(opencv-python、numpy、scipy 等),但什么都没有……我该如何解决这个问题?

错误:

File "C:\Users\Home\PycharmProjects\XG -I programming(trial part 2)\Mask_detection(trial 6).py", line 1, in <module>
    from imageai.Detection import VideoObjectDetection
  File "C:\Users\Home\PycharmProjects\XG -I programming(trial part 2)\venv\lib\site-packages\imageai\Detection\__init__.py", line 17, in <module>
    from imageai.Detection.YOLOv3.models import yolo_main, tiny_yolo_main
  File "C:\Users\Home\PycharmProjects\XG -I programming(trial part 2)\venv\lib\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\Home\PycharmProjects\XG -I programming(trial part 2)\venv\lib\site-packages\keras\layers\normalization\__init__.py)

代码:

from imageai.Detection import VideoObjectDetection
import cv2

camera = cv2.VideoCapture(0)
detector = VideoObjectDetection()

model_path = "./Models/yolo.h5"

detector.setModelTypeAsYOLOv3()
detector.setModelPath(model_path)
detector.loadModel()

while True:
ret, frame = camera.read()

rgb_frame = frame[:, :, ::-1]

video_detection = detector.detectObjectsFromVideo(camera_input=camera, output_file_path ="./camera_detected_video", frames_per_second=28, log_progress=True, minimum_percentage_probability=40)
cv2.imshow('frame', frame)

print(video_detection)

if cv2.waitKey(1) & 0xff == ord('q'):
    break


camera.release()
cv2.destroyAllWindows()
4

1 回答 1

0

问题出在keras版本上,尝试安装旧版本的keras

pip3 安装 keras==2.4.3

于 2022-02-05T06:03:51.277 回答