在实现 mRCNN 时，遇到“没有名为‘keras.engine’的模块”的问题

Question

我正在尝试实现 mRCNN，在尝试将其与 Keras 2.0 一起使用时，我多次遇到错误：

回溯（最近一次调用最后一次）： 文件“C:\Users\Warre\PycharmProjects est3\main.py”，第 3 行，位于 导入 mrcnn.model 文件“C:\Users\Warre\PycharmProjects\CricketAIMonitoring env\lib\site-packages\mask_rcnn-2.1-py3.10.egg\mrcnn\model.py”，第 23 行，位于 ModuleNotFoundError：没有名为“keras.engine”的模块

起来。

到目前为止，我纯粹尝试通过此链接获取该项目：https://github.com/EashanKaushik/car-damage-detecting-MaskRCNN/blob/main/README.md

功能，以及主 github 目录中带有气球的演示项目，但全面存在相同的问题。

提前感谢您的帮助！

Answer 1

mask_rcnn

软件包已过时。如果您需要模型，请从 torchvision 导入：

# Packages that I'm using
# torch Version: 2.3.0
# torch Version: 0.18.0

from torchvision.models.detection import MaskRCNN_ResNet50_FPN_Weights, maskrcnn_resnet50_fpn
import torch


# model = maskrcnn_resnet50_fpn(weights=MaskRCNN_ResNet50_FPN_Weights.DEFAULT)
model = maskrcnn_resnet50_fpn(weights=MaskRCNN_ResNet50_FPN_Weights.COCO_V1)

model.eval()
x = [torch.rand(3, 300, 400), torch.rand(3, 500, 400)]
predictions = model(x)