custom dataset

[zmj876902008]

Hello, thanks for your work. I want to know if I can use my own dataset training model?What would my dataset look like if I could?Thanks for your kind attention and look forward your prompt reply.

[hou-yz]

yes, you can try and use your own datasets. to this end, please create your own dataset class, or follow the ImageFolder class in torchvision.

hope this helps,
Yunzhong