WZMIAOMIAO / deep-learning-for-image-processing

deep learning for image processing including classification and object-detection etc.
GNU General Public License v3.0
23.29k stars 8.01k forks source link

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

Open QLW-07 opened 1 year ago

QLW-07 commented 1 year ago

使用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

zhGlade commented 8 months ago

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