Pytorch-UNet-master>utils>data_loading.py

模块，包   在package_runoob同级目录下，用test.py调用package_runoob包中内容

参考链接：

Python 模块 | 菜鸟教程 (runoob.com)

Dataset  抽象类，所有创建的数据集都需要进行子类化，并重写__len__和__getitem__

from torch.utils.data import Dataset

class BasicDataset(Dataset):

参考链接：

torch.utils.data - PyTorch中文文档 (pytorch-cn.readthedocs.io)

https://www.bilibili.com/video/BV1hE411t7RN?share_source=copy_web

assert()  宏，断言，程序在假设条件下，能够正常良好的运作

参考链接：

https://www.runoob.com/w3cnote/c-assert.html

splitext()  用于分离文件名与扩展名

import os

os.path.splitext(path)

e.g.

参考链接：

Python分离文件名与扩展名-Python splitext函数-嗨客网 (haicoder.net)

logging.info()  类似print（）的功能

参考链接：

python中logging模块上篇 - 知乎 (zhihu.com)

@staticmethod 用于修饰类中的方法,使其可以在不创建类实例的情况下调用方法

e.g.

class Time():

    @staticmethod
    def sec_minutes(s1,s2):

可以直接用，调用

Time.sec_minutes()

也可以创建实例，调用

t=Time()

t.sec_minutes()

参考链接：

简述python中的@staticmethod作用及用法 - 云+社区 - 腾讯云 (tencent.com)

Image.NEAREST 使用最近邻插值法调整img为newsize

Image.BICUBIC 使用双三插值法调整img为newsize

Image.resize()  单张图片大小变换

pil_img = pil_img.resize((newW, newH), resample=Image.NEAREST if is_mask else Image.BICUBIC)

np.asarray() 将结构数据转化为ndarray类型

 

ndim 返回数组维度ndim=len(img.shape)

import imghdr
import cv2
import numpy as np

img=cv2.imread("./00021_mask.png")
img=np.asarray(img)

print("img.ndim:",img.ndim)
print("img.shape",img.shape)
print("len(img.shape)",len(img.shape))

img.shape 输出数组的长、宽、通道数

np.transpose()  类似数组转置的操作，（0，1，2）对应（x，y，z）

arr.transpose((2,0,1))

相当于转成了（z, x, y)

numpy.load() 函数从具有npy扩展名(.npy)的磁盘文件返回输入数组

参考链接：

Python numpy.load()用法及代码示例 - 纯净天空 (vimsky.com)

Image.fromarray() array到image的转换

参考链接：

(17条消息) Image.fromarray的用法（实现array到image的转换）_weixin_39450145的博客-CSDN博客_image.fromarray

glob.golb() 查找符合特定规则的文件路径名

super()._init_() 自动继承父类属性

btw 整理Dataset--BasicDataset--CarvanaDataset关系

from torch.utils.data import Dataset

class BasicDataset(Dataset):
    def __init__(self, images_dir: str, masks_dir: str, scale: float = 1.0, mask_suffix: str = ''):
    def __len__(self):
    def __getitem__(self, idx):

class CarvanaDataset(BasicDataset):
    def __init__(self, images_dir, masks_dir, scale=1):
        super().__init__(images_dir, masks_dir, scale, mask_suffix='_mask')

Dataset 是torch中的一个抽象类，用于获取每一个数据及其label

BasicDataset是一个继承了Dataset的子类，需要重写__len__和__getitem__，这里的__init__是一种普通的继承关系，所以BasicDataset中一些属性需要自己定义、自己赋值

CarvanaDataset是一个继承了BasicDataset的子类，super().__init__自动继承父类属性，不需要对父类中定义过的属性再做定义了，可以用于对含_mask的后缀（suffix）实例化。