search

BR-IDL / PaddleViT

:robot: PaddleViT: State-of-the-art Visual Transformer and MLP Models for PaddlePaddle 2.0+
https://github.com/BR-IDL/PaddleViT
Apache License 2.0
1.22k stars 318 forks source link

训练自己的数据集 #111

Closed Q-amazing closed 2 years ago

Q-amazing commented 2 years ago

代码中说Dataset related classes and methods for ViT training and validation Cifar10, Cifar100 and ImageNet2012 are supported。 我想问下，我想训练自己的数据集需要自己重新写datase类吗？

xperzy commented 2 years ago

是的，因为我们无法得知您数据集的具体信息和使用方法，所以需要您实现自己的 Dataset 类，然后在项目中（如果是分类任务）dataset.py的 get_dataset 方法中添加您的Dataset class。