tensorflow 如何将图像文件转换为Tensor

ix0qys7i  于 2022-11-16  发布在  其他
关注(0)|答案(2)|浏览(198)

我正在尝试将NIH胸部X射线(kaggle中可用的数据集)图像转换为适合输入Keras模型的Tensor格式。

datagen=ImageDataGenerator(

)

datagen.fit('E:/Trial chest x-ray dataset/train resized (224,224)')

我尝试了上面提到的代码,但它是说-无法将字符串转换为浮点:'E:/试验胸部X射线数据集/调整大小的序列(224,224)'
然后我尝试从目录流

from keras.preprocessing.image import ImageDataGenerator

train_datagen=ImageDataGenerator(rescale=1./255)
valid_datagen=ImageDataGenerator(rescale=1./255)

train_generator=train_datagen.flow_from_directory(
        train_images,
        target_size=(224,224),
        batch_size=26,
        class_mode='categorical'
)

validation_generator=valid_datagen.flow_from_directory(
        validation_images,
        target_size=(224,224),
        batch_size=26,
        class_mode='categorical'
)

但它说-找到0图片属于0类。找到0图片属于0类。
有人能告诉我如何将图像转换为适合Keras模型的格式吗?

iyfjxgzm

iyfjxgzm1#

是否已创建这些文件夹?

  • 训练图像
  • 验证图像(_I)...

如果您还没有,请检查以下内容:https://www.tensorflow.org/hub/tutorials/tf2_image_retraining

jvlzgdj9

jvlzgdj92#

image = tf.io.read_file(image_path)
image = tf.image.decode_jpeg(image, channels=3)
image = tf.image.convert_image_dtype(image,tf.float32)
image = tf.image.resize(image,size = [img_size,img_size])

相关问题