文件名称:MCN-light-source-transfer:在PyTorch中正式实施“用于图像光源传输的多尺度自校准网络”(CVPRW 2021)
文件大小:2.09MB
文件格式:ZIP
更新时间:2024-05-08 14:25:56
Python
用于图像光源传输的多尺度自校准网络 团队:机智AI实验室 成员:王远志,陆涛,张彦铎,吴云涛 它包含了参加NTIRE 2021的代码:深度引导图像重新照明挑战赛第1条:一对一重新照明 先决条件 Linux(Ubuntu 1604或Windows 10) 水蟒 Python 3.7 NVIDIA RTX2080Ti GPU(11G或更大的内存)+ CUDA10.2 + cuDNN PyTorch 1.5.0 支配 科尔尼亚0.2.0 lpips-pytorch 入门 安装 创建一个conda虚拟环境 conda create -n MCN python=3.7 安装PyTorch conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.2 -c pytorch 安装主导 pip install domina
【文件预览】:
MCN-light-source-transfer-main
----checkpoints()
--------best_model()
----models()
--------ui_model.py(17KB)
--------cbam.py(6KB)
--------models.py(567B)
--------Modules_chutak.py(145KB)
--------base_model_DMSN.py(4KB)
--------base_model.py(3KB)
--------__pycache__()
--------networks_DMSN.py(27KB)
--------pix2pixHD_model.py(17KB)
--------models_lifeifei.py(7KB)
----test.py(4KB)
----dataset()
--------NTIRE2021_TEST()
----util()
--------util.py(4KB)
--------tvloss.py(711B)
--------html.py(2KB)
--------image_pool.py(1KB)
--------__pycache__()
--------visualizer.py(5KB)
--------pytorch_ssim()
----LICENSE(11KB)
----options()
--------__init__.py(1B)
--------base_options.py(6KB)
--------__pycache__()
--------test_options.py(1KB)
----README.md(2KB)
----data()
--------aligned_dataset.py(8KB)
--------base_dataset.py(3KB)
--------__pycache__()
--------custom_dataset_data_loader.py(943B)
--------data_loader.py(229B)
--------base_data_loader.py(195B)
--------image_folder.py(2KB)