Python
Python copied to clipboard
yolov2模型生成问题
我的Keras中的BatchNormalization不能够接收weights这个参数,没办法正常构建,以下是traceback截图,求助大佬www
图片里的代码信息不全。keras库可能有更新,这是最新的文档https://keras.io/api/layers/normalization_layers/batch_normalization/
图片里的代码信息不全。keras库可能有更新,这是最新的文档https://keras.io/api/layers/normalization_layers/batch_normalization/
谢谢您的回复,keras文档中的BatchNormalization类似乎已经没有了weights这个参数,在您提供的yolov2模型生成的yad2k.py文件中,实例化该类时输入了weights:这一参数导致程序无法正常执行,请问是否有新的解决方法呢
文件里的代码有点老了,而且不是我写的,有两种办法可以尝试下:
- 下载老版本的有weights的keras,在pypi上找一下21年左右发布的版本
- 现在版本有个
set_weights方法,改用这个方法设置权重而不是在初始化层的时候设置。
好的,谢谢您的回复