search

sunsmarterjie / SDL-Skeleton

A toolbox for object skeleton detection, can also be used for edge detection, building extraction and road extraction. TIP (2021)
119 stars 27 forks source link

datasets format for training #21

Open davidvct opened 1 year ago

davidvct commented 1 year ago

Hi, could you upload few sample of images and masks to github? I would like to train with my own custom datasets, but need to know the format of datasets.

I am not able to register for Baidu account to download the datasets.

davidvct commented 1 year ago

I am able to download the datasets files from google drive (from the link below):

https://github.com/sunsmarterjie/SDL-Skeleton/issues/12

StephenZhao1 commented 10 months ago

you can rewrite the pytorch dataset class, input is raw image and gt is mask png