tensorflow / models

Models and examples built with TensorFlow
Other
77.25k stars 45.75k forks source link

[OD][bug] models' loss doesn't use is_crowd #7378

Open netanel-s opened 5 years ago

netanel-s commented 5 years ago

System information

Describe the problem

It seems that when training Faster R-CNN or SSD model, their loss doesn't take into account is_crowd annotations (see RPN and classifier in Faster R-CNN, SSD). This means two problems:

  1. The single bounding box of the entire crowd is used as a positive example.
  2. The crowded objects inside the annotation will be assigned as negative examples, when naively they should be set as ignored examples.

These two problems will cause all models to be suboptimal or even bad at detecting objects in crowds.

ppwwyyxx commented 5 years ago

My implementation of Faster/Mask R-CNN handles crowd annotations. It does not use them as positive examples. And it can label anchors close to crowd region as ignored