数据集下载:https://www.kaggle.com/c/dogs-vs-cats-redux-kernels-edition
比较ViT,LeNet,VGG16在猫狗分类这个经典图像分类任务上的表现
整活代码,未进行调参,效果不足参考
懒得写requirements.txt了,确保你安装了PyTorch, Torchvision, vit-pytorch
省流:训练代码参见 demo_train.ipynb
进一步省流:推理结果参见 show.ipynb
数据集下载:https://www.kaggle.com/c/dogs-vs-cats-redux-kernels-edition
比较ViT,LeNet,VGG16在猫狗分类这个经典图像分类任务上的表现
整活代码,未进行调参,效果不足参考
懒得写requirements.txt了,确保你安装了PyTorch, Torchvision, vit-pytorch
省流:训练代码参见 demo_train.ipynb
进一步省流:推理结果参见 show.ipynb