我使用以下代码:
import tensorflow as tf
##############################################################
traindata = tf.keras.preprocessing.image.ImageDataGenerator(
rescale=1. / 255,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True)
input = traindata.flow_from_directory('VS/train')
modelo = tf.keras.Sequential()
modelo.add(tf.keras.layers.Conv2D(32, (3, 3),
activation=tf.keras.activations.relu))
modelo.add(tf.keras.layers.Flatten())
modelo.add(tf.keras.layers.Dense(64, activation=tf.keras.activations.relu))
modelo.add(tf.keras.layers.Dense(2, activation=tf.keras.activations.relu))
modelo.compile(loss='categorical_crossentropy', optimizer='rmsprop')
modelo.fit_generator(input, epochs=1)
Run Code Online (Sandbox Code Playgroud)
但是,我收到此错误:
所以通过运行下面的代码,我得到了这个错误
NotImplementedError: `fit_generator` is not yet enabled for unbuilt Model subclasses
Run Code Online (Sandbox Code Playgroud)
有人能告诉我什么是错的吗?
我想出了如何计算文件夹中的目录,但不知道如何编辑我的代码以递归计数子目录.任何帮助,将不胜感激.
到目前为止这是我的代码.
def nestingLevel(path):
count = 0
for item in os.listdir(path):
if item[0] != '.':
n = os.path.join(path,item)
if os.path.isdir(n):
count += 1 + nestingLevel(n)
return count
Run Code Online (Sandbox Code Playgroud)