torch_book.datasets.cv.div2k

Classes

PairedDIV2K

成对图片数据集

Module Contents

class torch_book.datasets.cv.div2k.PairedDIV2K

成对图片数据集

scale: int

HR_path: str | pathlib.Path

LR_path: str | pathlib.Path

__post_init__()

_check()

检查图片对是否匹配

__getitem__(index: int) → list[numpy.ndarray, numpy.ndarray]

加载(LR, HR)图片对

参数:

index -- 图片的索引

返回:

图片的二进制内容

返回类型:

buffer

__len__()

返回图片对数量