faster-rcnn.pytorch icon indicating copy to clipboard operation
faster-rcnn.pytorch copied to clipboard

Training Custom Dataset Tutorial

Open JiaLim98 opened this issue 4 years ago • 3 comments

To anyone who managed to compile this repo and train on custom datasets, can you please tell me the details to get it working? i.e. folder location, step by step linux commands, anaconda environment setup commands, etc.

I am struggling to get it working, if anyone that has a google colab notebook that compiles this repo and works perfectly, please let me know as well, I can try to follow all the steps and set it up on my local linux machine.

Many thanks!

JiaLim98 avatar Feb 23 '21 08:02 JiaLim98

You can try to checkout my implementation of this network with some abstractions for custom dataset.

My version is more recent, but last update was a long time ago.

loolzaaa avatar Feb 23 '21 09:02 loolzaaa

@JiaLim98 , I used to use this repository a lot for my work.. but it has since fallen into decline (I've also noticed pytorch 1.9.x works but 1.11.x complains when compiling this models cuda code so word of caution..)

the best I can give you right now is a link to the tutorials I wrote for this repos a while ago... BUT it's likely whatever you're running.. to need work.. when I get some time away from the day job.. I put in effort to just get this working again.. but the pytorch 1.11 nightly doesn't compile the cuda code.. I know C++ but this is not my day job and I may just release what I have just to help the pytorch community.. cuz this repo is really the only one doing anything that works for me.. even pytorchs own tutorials do not take in custom coco data for me..sadly

https://mahr.io/

EMCP avatar Oct 17 '21 10:10 EMCP

Hi @EMCP, thank you so much for the help! Much appreciated!

JiaLim98 avatar Oct 18 '21 09:10 JiaLim98