安装tensorflow,版本如下
pip uninstall tensorflow -y
pip uninstall keras -y
pip install tensorflow==2.4.3
pip install keras==2.4.0
在上面之后,会出现一些错误。您可以按照以下步骤解决它们。
@Error: [模块 'tensorflow' 没有属性 XXXXXXXX]
在model.py 或您的代码中,使用tf.compat.v1 解析一些api,例如tf.compat.v1.Session 或 import tensorflow.compat.v1 as tf
@Error: [ValueError: 试图将 'shape' 转换为张量并失败。错误:不支持任何值。]
mrcnn_bbox = KL.Reshape((-1, num_classes, 4), name="mrcnn_bbox")(x)
用这个 if-else 代码块替换:
if s[1]==None:
mrcnn_bbox = KL.Reshape((-1, num_classes, 4), name="mrcnn_bbox")(x)
else:
mrcnn_bbox = KL.Reshape((s[1], num_classes, 4), name="mrcnn_bbox")(x)
@Error: [ValueError: None values not supported.]
indices = tf.stack([tf.range(probs.shape[0]), class_ids], axis=1)
替换为
indices = tf.stack([tf.range(tf.shape(probs)[0]), class_ids], axis = 1)
@Error: [AttributeError: module 'keras.engine.saving' 没有属性 'load_weights_from_hdf5_group_by_name']
from keras import saving
替换为
from tensorflow.python.keras.saving import hdf5_format
和
saving.load_weights_from_hdf5_group(f, layers)
saving.load_weights_from_hdf5_group_by_name(f, layers)
替换为
hdf5_format.load_weights_from_hdf5_group(f, layers)
hdf5_format.load_weights_from_hdf5_group_by_name(f, layers)
参考: