使用预训练模型出现问题

[beathunder]

大佬，你好，我用你的源码尝试运行train.py时，当载入预训练模型时出现shape不匹配的问题，导致使用不了预训练模型，输入shape 为 416的和608的我都试过，都是这个问题，你可以帮看看吗

[bubbliiiing]

第一个conv就不对？预测呢？

[Alex-lubo]

我也出现这个问题。感觉从第一个conv就不对了，感觉工程代码是channel last，但是权重文件是chanel-first.

[bubbliiiing]

不是，这不影响

[Alex-lubo]

这个可能是tf的版本造成的, 我系统时tf2.0的，首先修改了一圈tensorflow.keras, 出现和@beathunder一样的问题。我首先加载时将load_weights时by_name和skip_mismatch都设置为false, 然后重新保存一下权重。然后修改class, 然后加载重新保存的权重文件时时设置by_name和skip_mismatch为true，everything works fine!

[bubbliiiing]

ok