🐞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 with keep_aspect_ratio_resizer, example:

image_resizer {
    fixed_shape_resizer {
    height: 640
    width: 640

👮‍♀️ You have to choose these values due to your model

Last updated