Datawhale AI 夏令营 CV 方向 Task 01 学习笔记

导入packages

import torch
torch.manual_seed(0)
torch.backends.cudnn.deterministic = False
torch.backends.cudnn.benchmark = True

import torchvision.models as models
import torchvision.transforms as transforms
import torchvision.datasets as datasets
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
from torch.autograd import Variable
from torch.utils.data.dataset import Dataset
import timm
import time

import pandas as pd
import numpy as np
import cv2
from PIL import Image
from tqdm import tqdm_notebook

train_label = pd.read_csv('phase1/trainset_label.txt')
val_label = pd.read_csv('phase1/valset_label.txt')

train_label['path'] = 'phase1/trainset/' + train_label['img_name']
val_label['path'] = 'phase1/valset/' + val_label['img_name']

模型训练与验证

加载模型

import timm
model = timm.create_model('resnet18', pretrained=True, num_classes=2)
model = model.cuda()

• 此处我更改timm版本：pip install timm==0.6.13

最后生成submit.csv并提交