Is it reliable to use TensorFlow (ML in general) to classify baggage bag tags based on the presence of a green stripe?
The images are identical except for the presence of the stripe on the side. I am trying to use a classify the images into 2 classes: greenStripe, noGreenStripe.
I tried to use tensorflow retrain with a small dataset (~40 pictures in each class and batch size of 8) but the results where really bad. I am afraid to commiting to training using more data as it is time consuming.
What do you suggest? Is there a better approach or does the problem lie in the small training dataset?
Topic tensorflow image-classification classification
Category Data Science