yolo2-pytorch
yolo2-pytorch copied to clipboard
How to use a custom dataset?
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 please do not use this code now, for the bug mentioned in this(https://github.com/longcw/yolo2-pytorch/issues/19)
Hello, Have you tried this? Is this code working now?
I also have doubts about how to write the dataloader. Any suggestions?
Instructions here