首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >tensorflow_datasets ImageFolder转换错误

tensorflow_datasets ImageFolder转换错误
EN

Stack Overflow用户
提问于 2022-02-22 23:47:20
回答 1查看 67关注 0票数 0

我有以下代码:

具有以下文件结构:

代码语言:javascript
运行
复制
recognition
  |_ train
      |_ 0
          |_ 0.bmp
          |_ 1.bmp
      |_ 1
          |_ 2.bmp
      |_ 2
          |_ 3.bmp
...

我想用下面的代码创建tensorflow图像数据集:

代码语言:javascript
运行
复制
import tensorflow_datasets as tfds

dataset_builder = tfds.folder_dataset.ImageFolder(recognition_data_path)
print(dataset_builder.info)

train_dataset = dataset_builder.as_dataset(split='train')

builder.info的输出如下:

代码语言:javascript
运行
复制
tfds.core.DatasetInfo(
    name='image_folder',
    full_name='image_folder/1.0.0',
    description="""
    Generic image classification dataset.
    """,
    homepage='https://www.tensorflow.org/datasets/catalog/image_folder',
    data_path='/kaggle/temp/recognition',
    download_size=Unknown size,
    dataset_size=Unknown size,
    features=FeaturesDict({
        'image': Image(shape=(None, None, 3), dtype=tf.uint8),
        'image/filename': Text(shape=(), dtype=tf.string),
        'label': ClassLabel(shape=(), dtype=tf.int64, num_classes=48),
    }),
    supervised_keys=('image', 'label'),
    disable_shuffling=False,
    splits={
        'test': <SplitInfo num_examples=unknown, num_shards=1>,
        'train': <SplitInfo num_examples=unknown, num_shards=1>,
    },
    citation="""""",
)

但是,当尝试创建train_dataset时,我会得到以下错误:

代码语言:javascript
运行
复制
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
/opt/conda/lib/python3.7/site-packages/tensorflow/python/framework/op_def_library.py in _apply_op_helper(op_type_name, name, **keywords)
    521                 as_ref=input_arg.is_ref,
--> 522                 preferred_dtype=default_dtype)
    523         except TypeError as err:

/opt/conda/lib/python3.7/site-packages/tensorflow/python/profiler/trace.py in wrapped(*args, **kwargs)
    162           return func(*args, **kwargs)
--> 163       return func(*args, **kwargs)
    164 

/opt/conda/lib/python3.7/site-packages/tensorflow/python/framework/ops.py in convert_to_tensor(value, dtype, name, as_ref, preferred_dtype, dtype_hint, ctx, accepted_result_types)
   1534           "Tensor conversion requested dtype %s for Tensor with dtype %s: %r" %
-> 1535           (dtype.name, value.dtype.name, value))
   1536     return value

ValueError: Tensor conversion requested dtype string for Tensor with dtype float32: <tf.Tensor 'args_0:0' shape=() dtype=float32>

During handling of the above exception, another exception occurred:

TypeError                                 Traceback (most recent call last)
/tmp/ipykernel_57/1360536814.py in <module>
....

我做错了什么?我在文档中没有发现任何可能的问题。

EN

回答 1

Stack Overflow用户

发布于 2022-02-23 08:49:50

好的,所以这个问题的解决方案是用.png而不是.bmp存储图像。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/71229718

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档