🐞
Model Debugging
🙄 Problems that we can face while training custom object detection
- 1.Model is not doing well on test set
- 2.Model is doing well on test set but doing bad on real world images
In case that model is not doing well on test set you can try one or more from the followings:
- Add dropout to
.config
file
box_predictor {
....
use_dropout: true
dropout_keep_probability: 0.8
....
}
- Replace
fixed_shape_resizer
withkeep_aspect_ratio_resizer
, example:
🖼 Before