After successfully submitting to dogs_vs_cats redux by working through lessons 1 and the dogs vs cats redux notebook I thought it would be a good learning experience to do it all over with my own small data set (and it really has been!)
The Issue I am having is that vgg.get_batches() is returning one more class than it should (I only have two classes, it tells me I have 3)
I have mimicked the directory setup
When I run…
#Fine tune the model
batches = vgg.get_batches(train_path, batch_size = batch_size)
val_batches = vgg.get_batches(valid_path, batch_size = batch_size*2)
vgg.model.optimizer.lr = 0.01
Found 34 images belonging to 3 classes.
Found 10 images belonging to 2 classes.
My “valid” directory and my “train” directory have identical folders in them.
Here is the path I am defining…
#Change the directory
#create relative path names
path = DATA_HOME_DIR + '/'
test_path = path + '/test/'
results_path = path + '/results/'
train_path = path + '/train/'
valid_path = path + ‘/valid/’
So I guess my question is, does vgg.get_batches determine the number of classes based on the number of folders in the directory? I looked at vgg16.py, tried to understand the get_batches, from there I went to Keras trying to track down my issue, but after a couple hours of troubleshooting around on this, I thought I would reach out here.)
Thanks for any help