项目场景:
测试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)进行加载模型。 当然也无法测试识别速度的影响了。