朴实无华的Dataset

最新推荐文章于 2022-05-26 23:05:11 发布

fuhao7i

最新推荐文章于 2022-05-26 23:05:11 发布

阅读量439

点赞数 1

分类专栏：一点一点向上爬🐌 文章标签： PyTorch Dataset 图像处理预处理数据加载

本文链接：https://blog.csdn.net/fuhao7i/article/details/120039153

版权

朴实无华的Dataset

自定义Dataset，该Dataset用于读取训练图像和标签图像，并做简单的尺寸调整和归一化处理，不包含其他的图像预处理操作。

具体的实例化和使用方法，可以参考main.py的使用。

SimpleDataset.py

import torch
import torch.nn as nn
import math
import torch.nn.functional as F
from PIL import Image
from torch.autograd import Variable
from torch.utils.data import DataLoader
from torch.utils.data.dataset import Dataset
from matplotlib.colors import rgb_to_hsv, hsv_to_rgb
import torchvision.transforms as transforms
import cv2
import os

def letterbox_image(image, size):
    iw, ih = image.size
    w, h = size
    scale = min(w/iw, h/ih)
    nw = int(iw*scale)
    nh = int(ih*scale)

    image = image.resize((nw,nh), Image.BICUBIC)
    new_image = Image.new('RGB', size, (0,0,0))
    new_image.paste(image, ((w-nw)//2, (h-nh)//2))
    return new_image, scale

def transform_small(params=None, grayscale