datasets.py

1. datasets.py 的头

import glob
import random
import os
import sys
import numpy as np
from PIL import Image
import torch
import torch.nn.functional as F

from utils.augmentations import horisontal_flip
from torch.utils.data import Dataset
import torchvision.transforms as transforms

2. 包含的函数

2.1. def pad_to_square(img, pad_value): # 图片填充到 方形分辨率 。 pad_value 值得是填补的值 0 填补黑色的点

1. 代码如下：

def pad_to_square(img, pad_value):  # 图片填充到 方形分辨率 。 pad_value 值得是填补的值 0 填补黑色的点
    c, h, w = img.shape
    dim_diff = np.abs(h - w)  # 高宽 之差
    # (upper / left) padding and (lower / right) padding
    pad1, pad2 = dim_diff // 2, dim_diff - dim_diff //