本项目为我给pytorch的dataloader（数据加载+预处理）提速的代码

具体内容参考博客《【pytorch】给dataloader提速：Kornia库 + GPU》

data_loaders_origin.py为来自STFAN的原始dateloader，data_loaders_kornia.py为我是用Kornia库提速后的dataloader。两个py文件均可直接运行，运行参数在config.py中。数据集可以是用DVD数据集，或者目录类似的数据集。目录格式为

.
├── test
│   ├── 720p_240fps_2
│   │   ├── GT
│   │   │   ├── 00000.jpg
│   │   │   ├── 00001.jpg
│   │   │   ├── 00002.jpg
│   │   │   └── ...
│   │   └── input
│   │       ├── 00000.jpg
│   │       └── ...
│   └── IMG_0003
│       ├── GT
│       │   └── ...
│       └── input
│           └── ...
└── train
    ├── 720p_240fps_1
    │   ├── GT
    │   └── input
    └── ...