Search code examples
tensorflowobject-detection-apifaster-rcnn

Tensorflow Faster R-CNN with Resnet-50 (v1) Object Detection API Results Only One Bounding Box on Per Image


Environment:

  • Google Colaboratory.
  • TPU.

Dataset:

  • Custom Dataset.
  • Number of classes=1

Results:

enter image description here

Expected Results:

enter image description here


Solution

  • While running the inference code change:

    'max_boxes_to_draw=5'    #max number of bounding boxes in the image
    

    You will get specified number of bounding boxes on image.