search

ArunMichaelDsouza / tensorflow-image-detection

A generic image detection program that uses Google's Machine Learning library, Tensorflow and a pre-trained Deep Learning Convolutional Neural Network model called Inception.
MIT License
328 stars 88 forks source link

Error #4

Closed SurenderMohanRaghav closed 6 years ago

SurenderMohanRaghav commented 6 years ago

File "retrain.py", line 817, in main class_count = len(image_lists.keys()) AttributeError: 'NoneType' object has no attribute 'keys'

ArunMichaelDsouza commented 6 years ago

Is the image training dataset provided in the specified manner ?

SurenderMohanRaghav commented 6 years ago

Yes, I provided.

ArunMichaelDsouza commented 6 years ago

I mean to say that the images need to be present in the training_dataset directory with folder names depicting the labels for individual training data sets. Only after the proper setup the transfer learning process will work using the command - $ bash train.sh.