Skip to content

How to train with custom dataset? #56

@panovr

Description

@panovr

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!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions