CV2_Task之数据增强和读取
前言
这次是对主要是对数据的增强操作和读取进行说明和如何用pytorch进行数据处理和进行数据读取。
赛题
赛题来源自Google街景图像中的门牌号数据集(The Street View House Numbers Dataset)
标题常见的数据处理方法
PIllow
以Lena的图片为例
然后进行pillow的一个函数处理
pillow操作可官方文档
CV2
可查看
官网
原图
cv的一步操作
标题对数据进行增广
torchvision
这是一个关于pytorch常见的数据处理的库,里面还有mnist,cifar等包的API
,也有常见的数据处理功能
import numpy as np
import torch
from PIL import Image
from torch.utils.data.dataset import Dataset
class SVHNDataset(Dataset):
def __init__(self, img_path, img_label, transform=None):
self.img_path = img_path
self.img_label =