Training on my own semantic dataset

[nikky4D]

Hi,

I'm new to this area and would like to train/fine tune on my own dataset.

Can you tell me how I can setup my own folder of images for training using main.py?

Thanks

[yang-song]

You will need to build a custom data loader in pytorch and replace the data loading pipeline in the provided code. You will need to modify the get_dataset function.