search

liyunsheng13 / BDL

MIT License
222 stars 30 forks source link

how to train own custom dataset？ #43

Open lucafei opened 3 years ago

lucafei commented 3 years ago

hi, i'm recently doing BDL, i want to ask, how to train own dataset?

liyunsheng13 commented 3 years ago

You just need to modify the DataLoader I wrote.

zhou-rui1 commented 3 years ago

Hi, I have only images and labels for binary seg,

how can I edit self.id_to_trainid = {3: 0, 4: 1, 2: 2, 21: 3, 5: 4, 7: 5, 15: 6, 9: 7, 6: 8, 16: 9, 1: 10, 10: 11, 17: 12, 8: 13, 18: 14, 19: 15, 20: 16, 12: 17, 11: 18}？

Any guidance will be so helpful!

liyunsheng13 commented 3 years ago

Maybe you no longer need self.id_to_trainid.