deep-learning-for-image-processing icon indicating copy to clipboard operation
deep-learning-for-image-processing copied to clipboard
verified publisher icon WZMIAOMIAO

TensorFlow 2.4.1使用mobilenetv3,自己训练的权重.ckpt转.h5，遇到错误

Open QLW-07 opened this issue 2 years ago • 1 comments

使用train_mobilenet_v3.py，进行的转换。网盘提供的权重可以转换，迁移学习的权重，如下错误：

Traceback (most recent call last): File "/home/xt/tf2.3-master/mobilenet_v3/trans_v3_weights.py", line 66, in main() File "/home/xt/tf2.3-master/mobilenet_v3/trans_v3_weights.py", line 60, in main weights = rename_var(ckpt_path, m_info) File "/home/xt/tf2.3-master/mobilenet_v3/trans_v3_weights.py", line 35, in rename_var assert k in var_dict, "{} not in var_dict".format(k) AssertionError: Conv/kernel not in var_dict

QLW-07 avatar Jul 12 '23 06:07 QLW-07

请问解决了吗 遇到一样的问题了

zhGlade avatar Mar 11 '24 09:03 zhGlade