Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to train Faster R-CNN on my own custom dataset and changing the RPN loss functions? #241

Open
meshghi opened this issue Aug 26, 2021 · 2 comments

Comments

@meshghi
Copy link

meshghi commented Aug 26, 2021

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.

@meshghi
Copy link
Author

meshghi commented Aug 27, 2021

Hello again,

I changed most of the code to work with my custom dataset. But, I'm stuck with handling the "difficult" trining objects. My custom dataset doesn't have any "difficult" object and thus I get errors during training in areas like below:

difficult.append(int(obj.find('difficult').text))

I was wondering how I could fix that? Thanks

@WuXiaoyu55
Copy link

Hello, I encountered the same problem, did you solve it?
difficult.append(int(obj.find('difficult').text))

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants