fasterrcnn-pytorch-training-pipeline

fasterrcnn-pytorch-training-pipeline copied to clipboard

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 ?

Jul 04 '22 09:07 TerrafYassin

@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.

Jul 04 '22 15:07 sovit-123