КЕРАС: AttributeError: объект 'BatchDataset' не имеет атрибута 'ndim', демонстрация сегментации изображения Tensorflow не работает на моей анаконде - PullRequest
Новая
       7

КЕРАС: AttributeError: объект 'BatchDataset' не имеет атрибута 'ndim', демонстрация сегментации изображения Tensorflow не работает на моей анаконде

0 голосов
/ 12 октября 2018

Я знаю, что есть проблема со строкой

history = model.fit(train_ds, steps_per_epoch=int(np.ceil(num_train_examples / float(batch_size))), epochs=epochs, validation_data=val_ds, validation_steps=int(np.ceil(num_val_examples / float(batch_size))), callbacks=[cp]) Я получаю сообщение об ошибке AttributeError: У объекта 'BatchDataset' нет атрибута 'ndim', и я использую демонстрацию tenorflows deg: сегментация изображения.Я думаю, что это связано с входными измерениями, но не могу найти выход из проблемы

информация для базы данных: 

train_ds = get_baseline_dataset(x_train_filenames,
                            y_train_filenames,
                            preproc_fn=tr_preprocessing_fn,
                            batch_size=batch_size)

val_ds = get_baseline_dataset (x_val_filenames, y_val_filenames, preproc_fn = val_preprocessing_fn, batch_size = batch_size)

def get_baseline_dataset (имена файлов, метки, preproc_fn = functools.partial (_augment), threads = 5, batch_size = batch_size, shuffle = True):

заранее спасибо (ps. работает в google colab, но не на anaconda / spyder)

...