I replaced your data with mines(same size and format) and set flag_multi_mask to True and change num_classes to 4 in train.py and change label names and colors in data.py but i keep getting this error while running train.py. Pls take a look
I used gg colab with Tensorflow-gpu==1.8.0, tensorflow==1.8, Keras==2.2.4
Detailed error:
KeyError Traceback (most recent call last)
in ()
48 validation_steps=10,
49 validation_data=valid_data,
---> 50 callbacks=[model_checkpoint,tb_cb])
51
52 # draw the loss and accuracy curve
14 frames
/usr/local/lib/python3.6/dist-packages/keras_preprocessing/image/utils.py in _list_valid_filenames_in_directory(directory, white_list_formats, split, class_indices, follow_links)
220 filenames = []
221 for root, fname in valid_files:
--> 222 classes.append(class_indices[dirname])
223 absolute_path = os.path.join(root, fname)
224 relative_path = os.path.join(
KeyError: 'image'
I replaced your data with mines(same size and format) and set flag_multi_mask to True and change num_classes to 4 in train.py and change label names and colors in data.py but i keep getting this error while running train.py. Pls take a look I used gg colab with Tensorflow-gpu==1.8.0, tensorflow==1.8, Keras==2.2.4 Detailed error:
KeyError Traceback (most recent call last)