BDL icon indicating copy to clipboard operation
BDL copied to clipboard

Published 20 hours ago • verified publisher icon liyunsheng13

how to train own custom dataset？

Open lucafei opened this issue 4 years ago • 3 comments

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

lucafei avatar Nov 14 '20 13:11 lucafei

You just need to modify the DataLoader I wrote.

liyunsheng13 avatar Nov 19 '20 19:11 liyunsheng13

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!

zhou-rui1 avatar Jun 02 '21 11:06 zhou-rui1

Maybe you no longer need self.id_to_trainid.

liyunsheng13 avatar Jun 03 '21 04:06 liyunsheng13