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

According to this article “https://medium.com/deepquestai/train-object-detection-ai-with-6-lines-of-code-6d087063f6ff”, 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