search

longcw / yolo2-pytorch

YOLOv2 in PyTorch
1.54k stars 421 forks source link

How to use a custom dataset? #27

Open alexiskattan opened 7 years ago

alexiskattan commented 7 years ago

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.

robotcator commented 7 years ago

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

AndresPMD commented 6 years ago

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

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

lematt1991 commented 6 years ago

Instructions here