如何自定义数据集

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

Open VEGETALOADING opened 1 year ago

VEGETALOADING commented 1 year ago

我希望在自己的数据集上尝试应用您的网络，请问哪里可以了解到您所使用的数据集如sklarge是如何标注的呢？

StephenZhao1 commented 1 year ago

直接标线段，生成mask就行，自己写一下dataset方法，inputs就是原图，target就是生成的mask