python使用神经网络拟合各种相机滤镜、胶片色彩

源码资源网

python使用神经网络拟合各种相机滤镜、胶片色彩
2.目录

• 【dir】static：权重文件和GUI资源文件
  
  • checkpoints：权重文件
  • src：资源文件
    
• 【dir】idea：理论研究
  
  • src：资源文件
    
• 【dir】pack：Nuitka打包文件
• 【dir】script：模型转换推理脚本
• train.py：训练脚本
• train_new.py：新的训练脚本
• dataset.py：数据构造
• gui.py：pyqt界面
• infer.py：推理脚本
• loss.py：损失函数
• models.py：模型
• preprocessing.py：数据预处理
• utils：功能函数
  

3.模型权重负片去色罩

• NegativeLabPro（NLP）：static/checkpoints/film-mask
  

奥林巴斯色彩模拟

• VIVID-浓郁色彩 ：static/checkpoints/olympus/vivid
• SoftFocus-柔焦 ：
• SoftLight-柔光 ：
• Nostalgia-怀旧颗粒 ：
• Stereoscopic-立体 ：
  
  

富士色彩模拟

• ACROS ：static/checkpoints/fuji/acros
• CLASSIC CHROME ：static/checkpoints/fuji/classic-chrome
• ETERNA ：static/checkpoints/fuji/enerna
• ETERNA BLEACH BYPASS ：static/checkpoints/fuji/eb
• CLASSIC Neg. ：static/checkpoints/fuji/classic-neg
• PRO Neg.Hi ：static/checkpoints/fuji/negstdhi
• NOSTALGIC Neg.：static/checkpoints/fuji/nostalgic-neg
• PRO Neg.Std ：static/checkpoints/fuji/negstd
• ASTIA ：static/checkpoints/fuji/astia
• PROVIA：static/checkpoints/fuji/provia
• VELVIA：static/checkpoints/fuji/velvia
• Pro 400H：static/checkpoints/fuji/pro400h
• Superia 400：static/checkpoints/fuji/superia400
• reala：static/checkpoints/fuji/rela
  
  

柯达色彩模拟

• Color Plus：static/checkpoints/kodak/colorplus
• Gold 200：static/checkpoints/kodak/gold200
• Portra 400：static/checkpoints/kodak/portra400
• Portra 160NC：static/checkpoints/kodak/portra160nc
• UltraMax 400：static/checkpoints/kodak/ultramax400
  
  

理光色彩模拟

• Std-标准 ：
• Vivid-鲜艳 ：
• Single-单色 ：
• SoftSingle-软单色 ：
• StiffSingle-硬单色 ：
• ContrastSingle-高对比对黑白 ：
• Neg-负片 ：
• R-Pos-正片 ：
• R-Nostalgia-怀旧 ：
• R-HDR-HDR ：
• R-Pos2Neg-正负逆冲 ：
  
  

宝丽来色彩模拟

• Polaroid：static/checkpoints/polaroid
  

索尼色彩模拟尼康色彩模拟佳能色彩模拟哈苏色彩模拟模型对应权重文件[td]

Model	Checkpoint
FilterSimulation	filmcnn.pth
UNet	unet.pth
UCM	best-ucm.pth

注：“使用unet模型获取更好的细节表现”

4.使用方法Pycharm等解释器（适用于所有平台）

• 配置环境：pip install -r requirements.txt
• python gui.py 使用GUI界面运行或者运行python infer.py 使用脚本推理
  

Windows GUI

• Windows链接
  

• 【V1.0】链接: https://pan.baidu.com/s/1WsBZbzCftyTMy3ZmzhJlDA 提取码: fmnq
• 【V1.1】链接: https://pan.baidu.com/s/1icLOXtVjUYqTkeDqf-o8Ag 提取码:e939
  

• 运行AIFilter.dist中的AIFilter.exe可执行文件
  

MacOS GUI

• Apple Silicon链接(M1/M2/m3)
  

• 【V1.0】链接: https://pan.baidu.com/s/1N5ux3eSUgYQTSB30iFw1GQ 提取码: nck8
• 【V1.1】链接: https://pan.baidu.com/s/1rnI5xPbwTkuZmetiWv0_6A 提取码: trbp
• 【V1.2】链接: https://pan.baidu.com/s/15v0pnFeGRMfCcVX5FE53_A 提取码: vp2x
• 【V1.3】链接: https://pan.baidu.com/s/1vFGzjBg0gy1eH3A1u6bC1w 提取码: qse4
  

• Intel链接(i5/i7/i9)
  

• 【V1.0】链接: https://pan.baidu.com/s/14afbEXX_C4F7b-OeFHXRQg 提取码: mjc7
• 【V1.1】链接: https://pan.baidu.com/s/1SmBLFE7MT4KwxzbSzpJGYA 提取码: nbue
  

• 运行AIFilter.app 或者拖到应用程序中
  

GUI介绍（全新设计）

python使用神经网络拟合各种相机滤镜、胶片色彩

顶部为标题为Film Simulate，上方新增不同品牌的切换，每个品牌对应不同的滤镜选择，左侧为滤镜选择区域，未选中的滤镜呈灰色状态，选中的滤镜呈彩色状态；底部提供了模型选择按钮，用户可通过切换不同的模型来推理图像。

iPhone推理

实现了将pytorch模型转换为coreml，在xcode上实现iPhone离线推理，12MP的图像推理时间为4s左右。但是目前没有开发app的想法（主要是不会，若有会ios开发的请联系我）

python使用神经网络拟合各种相机滤镜、胶片色彩

5.实验记录

1.参考idea/对比实验.md，框架如下：

python使用神经网络拟合各种相机滤镜、胶片色彩

2.参考idea/自适应图像色彩迁移方案.md，框架如下：

python使用神经网络拟合各种相机滤镜、胶片色彩

链接: https://pan.baidu.com/s/1uW-igwbAmO0U3Dbw8H-CuA

提取码下载：

文件名称:提取码下载.txt 
下载次数:0  文件大小:16 Bytes  售价:69金钱 [记录]
下载权限: 不限 [购买VIP]   [充值]   [在线充值]   【VIP会员6折；永久VIP4折】

安全检测，请放心下载