Is there a way to control the number of the predicted bounding boxes?

microsoft / scene_graph_benchmark

image scene graph generation benchmark

MIT License

380 stars 86 forks source link

Is there a way to control the number of the predicted bounding boxes? #67

Open JingbiaoMei opened 2 years ago

JingbiaoMei commented 2 years ago

With the originally released code, the number of the predicted bounding boxes is determined by each image. Is there a method that we could make sure the number of bboxes is consistent for all images? Currently, I am looking at the roi_heads.py and rpn.py but I got no luck.