How to specify IoU threshold and NMS threshold for prediction (Custom Object Detection model)

According to this article “”, I would like to set the IoU threshold and NMS threshold value when detecting objects (prediction) in the input image, because I need relatively fitted model. Does ImageAI support this requirement (custom detection) at this moment?

1 Like

Yes. In your detectObjectsFromImage funtion you can specify the following:

  • iou_threshold
  • object_threshold
  • nms_threshold