search

zhanghang1989 / ResNeSt

ResNeSt: Split-Attention Networks
https://arxiv.org/abs/2004.08955
Apache License 2.0
3.22k stars 494 forks source link

Training the segmentation model in your own data #137

Closed JJavierga closed 3 years ago

JJavierga commented 3 years ago

Hello. I have been looking for a way to train this model with my own segmentation data, but I only find the way to train several predefined datasets. Do I have to convert my dataset to the style of one of those available in order to train? Or is there an easier way? My dataset is composed of input images and output binary images for each channel.

Thanks

zhanghang1989 commented 3 years ago

That should be very similar to ADE20K dataset. Please check that out.