a-PyTorch-Tutorial-to-Image-Captioning icon indicating copy to clipboard operation
a-PyTorch-Tutorial-to-Image-Captioning copied to clipboard
verified publisher icon sgrvinod

How to make my own dataset

Open TalkIsCheap930 opened this issue 5 years ago • 2 comments

I want to make my own dataset to train this model, so how do I make my own dataset step by step?

TalkIsCheap930 avatar Nov 08 '20 08:11 TalkIsCheap930

I alread have collected 1800 images，I want to know how to make my own dataset to train this model

TalkIsCheap930 avatar Nov 08 '20 08:11 TalkIsCheap930

Go to utils.py which has the create_input_files function. Follow the steps carefully and you can train this on any dataset

darthgera123 avatar Nov 20 '20 11:11 darthgera123