night-echancement
下载源码
在github下载night-enhancement-main 源码进入:https://github.com/jinyeying/night-enhancement
下载权重进入
:https://www.dropbox.com/s/0ykpsm1d48f74ao/LOL params 0900000.pt?dl=0
放置在./results/LOL/model/
在PyCharm中打开项目night-enhancement-main切换虚拟环境
使用刚刚conda创建的night-enhancement 虚拟环境
安装依赖requirements.txt
(自己建一个)Terminal中安装依赖:pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
cycler==0.10.0
cython==0.29.32
guided-filter-pytorch==3.7.5
h5py==3.3.0
jsonpatch==1.32
jsonpointer==2.1
kiwisolver==1.3.1
kornia==0.5.8
lpips==0.1.4
matplotlib==3.4.2
networkx==2.6.1
numpy==1.24.1
opencv-python==4.5.1.48
pandas==1.3.0
protobuf==3.17.3
pypng==0.0.21
pytz==2021.1
pyyaml==5.4.1
scikit-image==0.18.2
scikit-video==1.1.11
scipy==1.6.0
tensorboardx==2.4
thop==0.0.31-2005241907
tifffile==2021.7.2
tqdm==4.61.2
运行main.py
把需要测试的图片放置在./LOL/
最后的结果在output文件夹下
可以比照一下最后训练好的模型