vedaseg icon indicating copy to clipboard operation
vedaseg copied to clipboard

How to train with custom dataset?

Open panovr opened this issue 4 years ago • 0 comments

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!

panovr avatar Nov 05 '21 07:11 panovr