Home Tuned Faster R-CNN model detect maximum 100 objects, but there is 300+ objects on the image. How to increase this maximum?

Questions

Tuned Faster R-CNN model detect maximum 100 objects, but there is 300+ objects on the image. How to increase this maximum?

byMR

November 25, 2023

I use this code to prepare model for optimization:

# load Faster RCNN pre-trained model
Faster_RCNN_tuned_model = torchvision.models.detection.fasterrcnn_resnet50_fpn(weights="DEFAULT")

# Get the number of input features 
in_features = Faster_RCNN_tuned_model.roi_heads.box_predictor.cls_score.in_features
    
# Define a new head for the detector with 2 classes (cell or fone)
Faster_RCNN_tuned_model.roi_heads.box_predictor = FastRCNNPredictor(in_features, 2) 
Faster_RCNN_tuned_model = Faster_RCNN_tuned_model.to(DEVICE)

and this code to load tuned model:

# Create Faster RCNN default model
best_Faster_RCNN_tuned_model = torchvision.models.detection.fasterrcnn_resnet50_fpn(weights="DEFAULT")
# Get the number of input features 
in_features = best_Faster_RCNN_tuned_model.roi_heads.box_predictor.cls_score.in_features
    
# Define a new head for the detector with 2 classes (cell or fone)
best_Faster_RCNN_tuned_model.roi_heads.box_predictor = FastRCNNPredictor(in_features, 2) 
# Load and setup parameters from saved best model
checkpoint = torch.load('C:\\temp\\datasets\\mediag\\models\\best_model.pth', map_location=DEVICE)
best_Faster_RCNN_tuned_model.load_state_dict(checkpoint['model_state_dict'])
best_Faster_RCNN_tuned_model = best_Faster_RCNN_tuned_model.to(DEVICE).eval()

There is 300+ objects on the images, but model detects only 100 objects:
Show objects count for first 5 images

How can I increase maximum objects count up to 3000 ?

May be I should manually setup this parameter ?

in_features = best_Faster_RCNN_tuned_model.roi_heads.box_predictor.cls_score.in_features
    
# Define a new head for the detector with 2 classes (cell or fone)
best_Faster_RCNN_tuned_model.roi_heads.box_predictor = FastRCNNPredictor(in_features, 2)

Please help me to increase detected objects count.