0
$\begingroup$

I have a couple of questions and I was wondering if you could answer them.

I have a bunch of images of the cars (side view only). I would like to train a model with those images. My objects of interest are 3 types of trucks that have different trailers. I rarely see two target objects at once in an image (maybe $1\over2$ in every 1000 images). However, I do see other types of cars that I do not want to detect.

My questions are:

Do you think I should tackle this problem as a detection task or classification task? (For example, should I consider multilabel classification or omit those pictures).

Should I also include other vehicles that I do not want to detect in my training dataset? Let's say that I do not assign a bounding box to them, but include them in training dataset just to make the system robust. (I trained YOLO with 200 images, sometimes the trained model confused and detected the wrong object that is not in any of classes, this happens when training with 2000 images per class? This is due to a small number of dataset or it is because of not including those images with no bounding boxes)?

Thank you in advance!

$\endgroup$
0
$\begingroup$

You have got nothing to lose by framing it as a multi-label detection task. All you have to do is remove the final Softmax layer and format your ground truth as multi-hot, and you can regress multiple labels directly. Mean Squared Error and Sum of Absolute Differences are neutrally biased with respect to false positives and false negatives if the data are balanced, whereas categorical cross-entropy is biased for higher recall and lower precision and usually does not work as well once the Softmax is removed, so choose your loss function accordingly.

Definitely include more objects in your training set that you don't want to detect--this will reduce overfitting by forcing the detection template(s) to improve their specificity.

| improve this answer | |
$\endgroup$

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.