New answers tagged

1 vote
Accepted

Can MSE be used for NN categorical classification problems

There are at least two reasons, why cross-entropy loss is preferred over mean squared error in classification problems. A theoretical reason Both aforementioned losses are negative logarithmic ...
0 votes

How do I train a model to classify if it's a Full Human Body in the picture?

First problem: I would start with training YOLOv5 model on two categories, full-body and partial-body. You would have to make your own dataset but since with yolo you usually use pretrained models, I ...
  • 1,195
0 votes

How to measure classification accuracy in object detection task?

For classification problems, using crossentropy and bce is always my go-to option. It usually performs best for these types of problems. However, using MSE or its variants is also common in some cases....
  • 101

Top 50 recent answers are included