上传者: 38629939
|
上传时间: 2021-10-31 17:21:37
|
文件大小: 56KB
|
文件类型: -
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