You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
In the folders train and validation there are 2 subfolders (OK and NOK), and both classes are well read by the generators. This is the output from them:
Found 40294 images belonging to 2 classes.
Found 1273 images belonging to 2 classes.
Setting classes=1 when declaring the model solves the error of the expected vs real shape of the loss vector but makes the classifier pointless.
Any idea about how to have more than 1 class when using generators?
The text was updated successfully, but these errors were encountered:
I am facing the following issue when trying to train a Squeezenet with no weights and two classes:
the code has been deeply inspired by this tutorial (https://blog.keras.io/building-powerful-image-classification-models-using-very-little-data.html)
In the folders train and validation there are 2 subfolders (OK and NOK), and both classes are well read by the generators. This is the output from them:
Setting classes=1 when declaring the model solves the error of the expected vs real shape of the loss vector but makes the classifier pointless.
Any idea about how to have more than 1 class when using generators?
The text was updated successfully, but these errors were encountered: