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 with custom dataset? #56

Open
panovr opened this issue Nov 5, 2021 · 0 comments
Open

How to train with custom dataset? #56

panovr opened this issue Nov 5, 2021 · 0 comments

Comments

@panovr
Copy link

panovr commented Nov 5, 2021

I want to train with custom dataset, and I have some questions:
(1) my custom dataset has two folders images and labels, where each label image is RGB image which uses different color for different object class. Should I organize this dataset in Pascal VOC format?
(2) I need to adapt voc_unet.py for custom dataset, Pascal VOC uses ignore_label for object boundary, how to set ignore_label for my own custom dataset?
(3) How to set crop_size_h, crop_size_w = 513, 513? My custom dataset has image dimension 512x512.
Thanks!

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

1 participant