狗品种识别
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
用户评论