Python icon indicating copy to clipboard operation
Python copied to clipboard

yolov2模型生成问题

Open jOHn1016CN opened this issue 1 year ago • 4 comments

我的Keras中的BatchNormalization不能够接收weights这个参数,没办法正常构建,以下是traceback截图,求助大佬www 屏幕截图 2024-08-13 012009

jOHn1016CN avatar Aug 12 '24 17:08 jOHn1016CN

图片里的代码信息不全。keras库可能有更新,这是最新的文档https://keras.io/api/layers/normalization_layers/batch_normalization/

Dragon-GCS avatar Aug 13 '24 08:08 Dragon-GCS

图片里的代码信息不全。keras库可能有更新,这是最新的文档https://keras.io/api/layers/normalization_layers/batch_normalization/

谢谢您的回复,keras文档中的BatchNormalization类似乎已经没有了weights这个参数,在您提供的yolov2模型生成的yad2k.py文件中,实例化该类时输入了weights:这一参数导致程序无法正常执行,请问是否有新的解决方法呢

jOHn1016CN avatar Aug 14 '24 17:08 jOHn1016CN

文件里的代码有点老了,而且不是我写的,有两种办法可以尝试下:

  1. 下载老版本的有weights的keras,在pypi上找一下21年左右发布的版本
  2. 现在版本有个set_weights方法,改用这个方法设置权重而不是在初始化层的时候设置。

Dragon-GCS avatar Aug 15 '24 10:08 Dragon-GCS

好的,谢谢您的回复

jOHn1016CN avatar Aug 20 '24 12:08 jOHn1016CN