1
$\begingroup$

I'm trying to retrain a neural network using transfer learning that can classify whether an image has a certain object, say, a car. My positive sample dataset is quite small, only 2500~ images. It works really well with "regular" binary classification (2500 images of cars/2500 images of flowers and it has to differentiate between those two) but the problem is that I'm not sure how to make it classify for all types of images, or how to make it 2500 images of cars/2500 random images and it'd have to classify whether Yes - the image is a car or No - it's not a car. What would those random images be? Thanks!

$\endgroup$
0
$\begingroup$

Have you tried creating a dataset with 2500 images of cars and ~2500 images of random stuff and trained it first? if so you should try that and see how it works, if it doesnt work you might have to add more images in the "distractor" class. Provide more information about what you find after you do this experiment and may be we can help further?

| improve this answer | |
$\endgroup$
  • $\begingroup$ I did! Sorry for not mentioning that. I downloaded batch images of nature, dogs, food, etc. I think that might have been the problem? The images that I downloaded could have been too similar to each other so it would have ended up being a bunch of batches of similar images. It failed, miserably. It thought that a hotdog was a car. Basically ended up having a very high bias towards the positive examples. That's where I got lost and started doubting if it's even possible to do with such a small dataset of positive examples. $\endgroup$ – 123 Feb 19 at 13:04

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Not the answer you're looking for? Browse other questions tagged or ask your own question.