prabindh / ctsoc-ewaste

Object detection for e-waste
4 stars 3 forks source link

Segregation of merged boxes - NMS #1

Closed prabindh closed 3 years ago

prabindh commented 3 years ago

In regard to merged bounding boxes - I mentioned about NMS, in relation to merging of many bounding boxes for detection in one of the images - you can see this config, in the region in the colab script below:

post_processing {
  batch_non_max_suppression {
    score_threshold: 1e-8
    iou_threshold: 0.5
    max_detections_per_class: 100
    max_total_detections: 100
  }

Also refer to discussions on "batch_non_max_suppression" parameter - ex, https://stackoverflow.com/questions/57311222/limitation-of-number-of-predictions-in-tensorflow-object-detection-api