Ошибка с сверточной сетью кошек и собак: model.fit_generator - PullRequest
Новая
       4

Ошибка с сверточной сетью кошек и собак: model.fit_generator

1 голос
/ 28 февраля 2020

Я использую PDF для создания своей первой сверточной сети, используя кошек и собак, и у меня постоянно возникает ошибка. Текст: ПРЕДУПРЕЖДЕНИЕ: tenorflow: режимы sample_weight приведены с ... до
['...'] ПРЕДУПРЕЖДЕНИЕ: tenorflow: режимы sample_weight приведены с ... до
['...']

Соответствующий код вставлен в два раздела ниже. Любая помощь будет признательна, потому что я бью стену в отношении этого.

Этот верхний бит работает, но может иметь отношение: 

#Build the network

#Import needed layers and models from tensorflow.keras
import tensorflow as tf
from tensorflow.keras.layers import Dense, Activation, Flatten, Conv2D, MaxPooling2D, Dropout
from tensorflow.keras.models import Sequential

#Build model--Use sequential value--Most common
model = models.Sequential()

#Input layer
model.add(layers.Conv2D(32, (3,3), activation = 'relu', 
                        input_shape = (150, 150, 3)))          
model.add(layers.MaxPooling2D(2,2))

#First hidden layer
model.add(layers.Conv2D(64, (3,3), activation = 'relu'))
model.add(layers.MaxPooling2D(2,2))

#Second hidden layer
model.add(layers.Conv2D(128, (3,3), activation = 'relu'))
model.add(layers.MaxPooling2D(2,2))

#Third hidden layer
model.add(layers.Conv2D(128, (3,3), activation = 'relu'))
model.add(layers.MaxPooling2D(2,2))

#Fourth hidden layer
model.add(layers.Flatten())
model.add(layers.Dense(512, activation = 'relu'))

#Output layer
model.add(layers.Dense(1, activation = 'sigmoid'))

#
from tensorflow.keras import optimizers

#Compilation step
model.compile(loss = 'binary_crossentropy',
             optimizer= 'adam',
             metrics=['acc'])

#Read images from directories
from tensorflow.keras.preprocessing.image import ImageDataGenerator

train_datagen = ImageDataGenerator(rescale = 1./255)

test_datagen = ImageDataGenerator(rescale = 1./255)

train_generator = test_datagen.flow_from_directory(
        train_dir,
        target_size = (150, 150),
        batch_size = 20,
        class_mode = 'binary')

validation_generator = test_datagen.flow_from_directory(
        validation_dir,
        target_size = (150, 150),
        batch_size = 20,
        class_mode = 'binary')

Подходит модель с генератором партии Эта часть кода - это то, что вызывает историю ошибок. = 3 и использует базу данных kagglecatsanddogs от Microsoft

1 Ответ

0 голосов
/ 17 марта 2020

Следующее предупреждение исправлено в ночной версии Tensorflow и будет включено в следующую стабильную версию, Tensorflow 2.2

WARNING:tensorflow:sample_weight modes were coerced from ... to
['...'] WARNING:tensorflow:sample_weight modes were coerced from ... to
['...']

В настоящее время, чтобы сделать работу, пожалуйста, установите Tensorflow Nightly Version, как показано ниже: 

!pip install tf-nightly

Для получения более подробной информации, пожалуйста, обратитесь к этому Github Issue .

...