微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

AttributeError: 模块“tensorflow._api.v2.io”没有属性“tf_record_iterator”

如何解决AttributeError: 模块“tensorflow._api.v2.io”没有属性“tf_record_iterator”

我有一个名为 train00.tfrecord文件,我想通过 tensorflow 读取它。所以我写了下面的代码

frame_lvl_record = "frame-sample/frame/train00.tfrecord"
for example in tf.io.tf_record_iterator(frame_lvl_record):
    print(example)

但它返回了以下错误

AttributeError                            Traceback (most recent call last)
/tmp/ipykernel_20529/2760031393.py in <module>
----> 1 for example in tf.io.tf_record_iterator(frame_lvl_record):
      2     print(example)

AttributeError: module 'tensorflow._api.v2.io' has no attribute 'tf_record_iterator'

我该如何解决这个依赖问题?

解决方法

'''使用这个命令读取TFRecordDataset

data=tf.data.TFRecordDataset("filenames")                     
n=10          
for raw_record in data.take(n):                              
    print(repr(raw_record))      

'''

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。