文件名称：MASA-SR:MASA-SR
文件大小：33KB
文件格式：ZIP
更新时间：2024-04-21 07:50:18
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)