facebookresearch / Detectron

FAIR's research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.
Apache License 2.0
26.21k stars 5.45k forks source link

How to train Faster R-CNN on my own custom dataset? #1028

Open meshghi opened 2 years ago

meshghi commented 2 years ago

Hello. I want to:

1- train Faster RCNN on my own custom dataset. 2- use the pre-trained Faster RCNN on the VOC2007 as the initial weights to train it then on my own custom dataset. 3- modify the RPN network with different loss functions that I have come up with.

I will greatly appreciate it if you could help me/let me know how to achieve the above goals.

under-score commented 2 years ago

Have a look at towardsdatascience.com for custom object detection.

s4nyam commented 1 year ago

Still Looking for some productive references, not literature, but proper guidelines to code. Thanks :)

aymuos15 commented 11 months ago

https://colab.research.google.com/drive/1oFxPpBeE8SzSQq7BTUv28IIqQeiHHLdj?usp=sharing#scrollTo=e1BoCmE3j54d

This link seemed fairly helpful (for question 1). At least provides some form of intuition as to how structure your pipeline.

s4nyam commented 11 months ago

Thanks, I finished the project, its called as DeepSeaNet, have some click to run all notebooks along with that, try looking - https://arxiv.org/abs/2306.06075