Change scale and aspect ratio of RPN network

sovit-123 / fasterrcnn-pytorch-training-pipeline

PyTorch Faster R-CNN Object Detection on Custom Dataset

MIT License

223 stars 75 forks source link

Change scale and aspect ratio of RPN network #17

Open TerrafYassin opened 2 years ago

TerrafYassin commented 2 years ago

Hi , I need to modify the aspect ratio and scale to cover small object detection problem , and i see in your project you import RPN network from Torchvision models , how can i modify RPN network ?

sovit-123 commented 2 years ago

@TerrafYassin You can take a look at this model => https://github.com/sovit-123/fastercnn-pytorch-training-pipeline/blob/main/models/fasterrcnn_mini_darknet_mini_head.py It contains custom head and Predictor. You will get an idea of how to accomplish what you want.