hey @Saurav I would recommend that your reduce the input size and also the backbone to resnet50.

I had good performances with an input size 512x512 and resnet50.