【发布时间】:2021-09-19 00:32:03
【问题描述】:
需要帮助来解决这个问题。我不确定出了什么问题,但错误仍然存在。环顾四周,但找不到类似的问题。
import matplotlib.pyplot as plt
from PIL import Image
import os
import numpy as np
from skimage import io
from keras.preprocessing.image import ImageDataGenerator
from matplotlib import cm
from mpl_toolkits.axes_grid1 import ImageGrid
import math
%matplotlib inline
import keras
import tensorflow as tf
from keras.models import Model
batch_size=32
datagen_args = dict(rotation_range=20,
width_shift_range=0.2,
height_shift_range=0.2,
rescale=1./255)
datagen = ImageDataGenerator(**datagen_args)
train_datagenerator = datagen.flow_from_directory('/content/drive/MyDrive/cats_dogs_small/train',target_size=(128,128),
batch_size=batch_size,interpolation="lanczos",shuffle=True)
valid_datagenerator = datagen.flow_from_directory('/content/drive/MyDrive/cats_dogs_small/validation',target_size=(128,128),
batch_size=batch_size,interpolation="lanczos",shuffle=True)
epochs = 25
hist = Model.fit_generator(train_datagenerator,
steps_per_epoch= math.ceil(train_datagenerator.samples//batch_size),
epochs=epochs, validation_data=valid_datagenerator, validation_steps=math.ceil(valid_datagenerator.samples//batch_size),verbose = 1, workers=8)
错误信息如下:
TypeError Traceback (most recent call last)
<ipython-input-69-178574fd407f> in <module>()
2 hist = Model.fit_generator(train_datagenerator,
3 steps_per_epoch= math.ceil(train_datagenerator.samples//batch_size),
----> 4 epochs=epochs, validation_data=valid_datagenerator, validation_steps=math.ceil(valid_datagenerator.samples//batch_size),verbose = 1, workers=8)
TypeError: fit_generator() missing 1 required positional argument: 'generator'
【问题讨论】:
-
Model 是一个类,你的模型保存在哪个变量中?
-
您说您需要帮助,但随后忽略了人们对您问题的澄清?在您定义模型在此代码中的位置之前,这是无法真正回答的。
-
抱歉,回复晚了!我昨天才设法再次尝试。我现在已经想通了。谢谢!
标签: python image tensorflow keras google-colaboratory