Sovit Ranjan Rath

You are inputting a 2D tensor to the model whereas it should be a 4D tensor, that is, [batch size, channels, height, width].

Hi. The best place to make these changes would be in the `datasets.py` file. Also, none of the augmentations will be applied by default unless you pass the `--use-train-aug` argument...

You can try Faster RCNN MobileNet V3 model.

Try lowering the initial learning rate to 0.00001

Hi, this generally does not happen. This is one of the first times I have seen this issue. Were you able to solve the issue?

Ok. May I know your PyTorch version. I will try to debug deeper this weekend. Also, if possible is there a link to your dataset that I can use for...

I compute the FPS on the forward pass only. So, the cv2 functions won't affect it. Faster RCNN, although accurate, struggles a bit to give high FPS on edge devices....

I think you can use the Faster RCNN ResNet50 FPN V2 model. It works very well for small objects.

May I know the dataset.

Have you tried with the default code in the repository without additional changes to the image processing techniques? If so, can you please let me know the result? Also, for...