faster-rcnn.pytorch
faster-rcnn.pytorch copied to clipboard
Training Custom Dataset Tutorial
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!
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.
@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/
Hi @EMCP, thank you so much for the help! Much appreciated!