我只是想知道tensorflow模型(h5)文件的tensorflow版本。
我想知道tensorflow的版本号
这样的东西对你有用吗?
import h5py
import re
def get_tensorflow_version(h5):
with h5py.File(h5, 'r') as f:
metadata = f.attrs['model_config'].decode('utf-8')
version = re.search(r'"tensorflow_version":s*"([^"]*)', metadata)
if version:
return version.group(1)
else:
return "No Version"
print(get_tensorflow_version('xxx.h5'))
将xxx替换为文件名