Kaggle竞赛中的犬种识别挑战,比赛的网址是https://www.kaggle.com/c/dog-breed-identification 在这项比赛中,尝试确定120种不同的狗。该比赛中使用的数据集实际上是著名的ImageNet数据集的子集。
基本思路
加载自定义数据集
微调ResNet18模型
训练模型
基于pytorch的代码
日常导入需要用到的python库
import torch
import torch.nn as nn
import torch.optim as optim
import torchvision
from torchvision import transf
1