yolo2-pytorch icon indicating copy to clipboard operation
yolo2-pytorch copied to clipboard

How to use a custom dataset?

Open alexiskattan opened this issue 6 years ago • 3 comments

I would like to retrain the network on a custom dataset of food items. I'm sort of new to this. Are there instructions about how to do it? I am looking at torch.utils.data.Dataset but am not sure how it links to the rest of the program.

alexiskattan avatar Jun 26 '17 23:06 alexiskattan

@alexiskattan please do not use this code now, for the bug mentioned in this(https://github.com/longcw/yolo2-pytorch/issues/19)

robotcator avatar Jun 29 '17 07:06 robotcator

Hello, Have you tried this? Is this code working now?

I also have doubts about how to write the dataloader. Any suggestions?

AndresPMD avatar Oct 18 '17 07:10 AndresPMD

Instructions here

lematt1991 avatar Oct 24 '17 13:10 lematt1991