【文件属性】:
文件名称:MASA-SR:MASA-SR
文件大小:33KB
文件格式:ZIP
更新时间:2021-04-01 14:03:38
Python
MASA-SR
我们的CVPR2021论文PyTorch正式实现
依存关系
Python3
火炬> = 1.1.0
火炬视觉> = 0.4.0
准备数据集
下载和
将数据集放置在以下结构中: CUFED
├── train
│ ├── input
│ └── ref
└── test
└── CUFED5
开始使用
克隆此仓库git clone https://github.com/Jia-Research-Lab/MASA-SR.git
cd MASA-SR
下载数据集。 根据您的数据路径,在test.py和train.py修改参数--data_root 。
评估
下载预训练的模型并将其放入pretrained_weights/文件夹中
可以从下载预先训练的模型
masa_rec.pth :仅受重建损失训练
masa.pth :受过所有损失的训练
运行test.
【文件预览】:
MASA-SR-main
----train.py(6KB)
----utils()
--------util.py(3KB)
--------calculate_PSNR_SSIM.py(6KB)
----models()
--------archs()
--------__init__.py(29B)
--------modules.py(3KB)
--------losses.py(21KB)
--------vgg_model.py(6KB)
--------trainer.py(19KB)
----test.py(3KB)
----flop_count.py(16KB)
----train.sh(542B)
----dataloader()
--------data_sampler.py(2KB)
--------dataset.py(15KB)
--------__init__.py(1KB)
----test.sh(1KB)
----README.md(2KB)