python - 使用 `keras.utils.Sequence` 作为输入时,不支持 `y` 参数
问题描述
我正在使用监督学习。我没有对任何我试图查看是否可以通过提供高分辨率输出来改变输入分辨率的东西进行分类。我有低分辨率输入和高分辨率输出。输出也是图像而不是类名。输出是否应该作为列表给出。
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import tensorflow as tf
from keras.models import Sequential
from tensorflow.keras.layers import Conv2D, MaxPooling2D
from tensorflow.keras.layers import Input, Dropout, Flatten, Dense
from keras.layers import Convolution2D
from keras.preprocessing.image import ImageDataGenerator
from keras.layers import UpSampling2D
from keras.models import Model
from tensorflow.keras.layers import BatchNormalization
test_datagen = ImageDataGenerator(rescale=0./255)
train_datagen = ImageDataGenerator(rescale=0./255)
train_out_datagen = ImageDataGenerator(rescale=0./255)
validation_datagen = ImageDataGenerator(rescale=0./255)
train_indir = r"D:\\UAV\\Train\\Input"
validation_indir = r"D:\\UAV\\Train\\validation"
output_outdir = r"D:\\UAV\\Train\\Output"
input_gen = train_datagen.flow_from_directory(
train_indir,
target_size = (3402,3401),#(3402,3401),
batch_size =32,
color_mode='rgb',
class_mode='input')
valid_gen = validation_datagen.flow_from_directory(
validation_indir,
target_size = (3402,3401), #(3402,3401),
batch_size =32,
color_mode='rgb',
class_mode = 'input')
output_gen = train_out_datagen.flow_from_directory(
output_outdir,
target_size = (3402,3401),#(3402,3401),
batch_size =32,
color_mode='rgb',
class_mode='input')
base_model = tf.keras.applications.ResNet50(
include_top=False,
weights="imagenet",
input_shape=(3402,3401,3),
pooling=None,
)
for layer in base_model.layers[:]:
layer.trainable = False
model = Sequential()
model.add(base_model)
model.add(Convolution2D(3,9,activation='relu',padding='same'))
model.add(UpSampling2D())
model.add(UpSampling2D())
model.add(BatchNormalization())
model.add(Convolution2D(3,9,activation='relu',padding='same'))
model.build((None, 3402, 3401, 3))
model.summary()
model.compile(optimizer="adam", loss='mean_squared_error', metrics=['mean_squared_error'])
model.fit(input_gen,output_gen,validation_data = valid_gen,batch_size =32,epochs=100)
解决方案
推荐阅读
- sql - Sql Server 删除在哪里
- typescript - 单击离子上的按钮后禁用复选框
- haskell - 尝试安装“idris”:“hackage.haskell.org”的包列表不存在。运行 'cabal update' 下载它
- angular - 如何在Angular中动态更改绑定变量?
- c++ - 通过qt TCP/IP发送图片
- google-analytics - 在 Google Analytics(分析)中为 Google 表格合并目标计数
- node.js - 如何使用 TypeScript 正确键入 Sequelize JOIN?
- python - 在烧瓶中绘制带有下拉菜单的离线图表
- java - Hibernate OneToMany 关系是 PersistentBag 而不是 List
- python - Beautiful Soup 在子元素中找到 href