ValueError: Unknown layer: Mish

it2024-12-19  13

项目场景:

测试yolov4有网络结构的模型和无网络结构的模型是识别速度是否有影响


问题描述:

下面这行代码报错:ValueError: Unknown layer: Mish

yolo_model = load_model(model_path, compile=False)

原因分析:

yolov4 中用到了个人构建的Mish函数,keras库中是没有的,当在使用self.yolo_model = load_model(model_path, compile=False),load_model进行加载模型时,虽然模型中保存了模型的结构,但是因为keras库中没有mish函数,所以会报错。。


解决方案:

应该用self.yolo_model.load_weights(self.model_path)进行加载模型。 当然也无法测试识别速度的影响了。
最新回复(0)