Skip to content

generate any filter like fuji/ricoh/olympus/sony/canon/nikon freely by neural network!

License

Notifications You must be signed in to change notification settings

fg0521/filter4free

Repository files navigation

FilterSimulation

1.介绍

使用神经网络拟合各种相机滤镜胶片色彩!后续会一直更新迭代下去,欢迎关注,欢迎右上角点一下Started☆!这会有很大的帮助,谢谢!也欢迎白嫖!

2.目录

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

3.模型权重

负片去色罩
  1. NegativeLabPro(NLP):static/checkpoints/film-mask/best.pth ✅️
奥林巴斯色彩模拟
  1. VIVID-浓郁色彩static/checkpoints/olympus/vivid/best.pth
  2. SoftFocus-柔焦 :❎
  3. SoftLight-柔光 :❎
  4. Nostalgia-怀旧颗粒 :❎
  5. Stereoscopic-立体 :❎
富士色彩模拟
  1. ACROS :❎
  2. CLASSIC CHROMEstatic/checkpoints/fuji/classic-chrome/best.pth
  3. ETERNA :❎
  4. ETERNA BLEACH BYPASS :❎
  5. CLASSIC Neg. static/checkpoints/fuji/classic-neg/best.pth
  6. PRO Neg.Hi :❎
  7. NOSTALGIC Neg.static/checkpoints/fuji/nostalgic-neg/best.pth
  8. PRO Neg.Std :❎
  9. ASTIA :❎
  10. PROVIAstatic/checkpoints/fuji/provia/best.pth
  11. VELVIAstatic/checkpoints/fuji/velvia/best.pth
  12. Pro 400Hstatic/checkpoints/fuji/pro400h/best.pth
  13. Superia 400static/checkpoints/fuji/superia400/best.pth
柯达色彩模拟
  1. Color Plusstatic/checkpoints/kodak/colorplus/best.pth
  2. Gold 200static/checkpoints/kodak/gold200/best.pth
  3. Portra 400static/checkpoints/kodak/portra400/best.pth
  4. Portra 160NCstatic/checkpoints/kodak/portra160nc/best.pth
  5. UltraMax 400static/checkpoints/kodak/ultramax400/best.pth
理光色彩模拟
  1. Std-标准 :❎
  2. Vivid-鲜艳 :❎
  3. Single-单色 :❎
  4. SoftSingle-软单色 :❎
  5. StiffSingle-硬单色 :❎
  6. ContrastSingle-高对比对黑白 :❎
  7. Neg-负片 :❎
  8. R-Pos-正片 :❎
  9. R-Nostalgia-怀旧 :❎
  10. R-HDR-HDR :❎
  11. R-Pos2Neg-正负逆冲 :❎
宝丽来色彩模拟
  1. Polaroidstatic/checkpoints/polaroid/best.pth
索尼色彩模拟
尼康色彩模拟
佳能色彩模拟
哈苏色彩模拟
模型对应权重文件
Model Checkpoint
FilterSimulation best-v4
UNet best.pth
UCM best-ucm.pth

4.使用方法

Pycharm等解释器(适用于所有平台)
  1. 配置环境:pip install -r requirements.txt
  2. python gui.py 使用GUI界面运行 或者 ython infer.py 使用脚本推理
Windows GUI
  1. Windows链接
  1. 运行AIFilter.dist中的AIFilter.exe可执行文件
MacOS GUI
  1. Apple Silicon链接(M1/M2/m3)
  1. Intel链接(i5/i7/i9)
  1. 运行AIFilter.app 或者拖到应用程序
GUI介绍

实验记录

1.参考idea/对比实验.md,框架如下:

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

用户提问★★★★★

  • MacOS M系列出现已损坏,无法打开提示
    1. 打开电脑终端,输入sudo spctl --master-disable密码,回到【系统偏好设置】【安全性与隐私】【安全性】/【通用,选中任何来源选项。
    2. 打开 “终端的完全磁盘访问” 权限。【系统设置】【隐私与安全性】【完整磁盘访问权限】 勾选了【终端】 (如果之前勾选过-然后+)
    3. 打开【系统设置】【隐私与安全性】【安全性】板块下,不明身份的开发者 点下方的【仍要打开】
    4. 先把app放到【应用程序】再在【终端】执行sudo xattr -rd com.apple.quarantine 你的app路径,例如:sudo xattr -rd com.apple.quarantine /Applications/FilterSimulation.app
  • MacOS M系列出现闪退
  1. 可以在terminal中使用open AIFilter.app 命令打开或右键显示包内容ContentsMacOSAIFilter右键在终端中打开
  2. 安装command-line-tools:在【终端】中运行xcode-select --install命令进行安装,安装完成后输入sudo codesign --force --deep --sign - 你的app路径 ,例如:sudo codesign --force --deep --sign - /Applications/FilterSimulation.app
  • 其他相机/手机使用后是否存在偏差
  • 关于运行源码问题
    1. M系列芯片在运行gui.py时候,可能出现PyQt5的安装问题,由于目前该库不支持使用pip安装,需要使用berw安装,再将这个库拖入到自己的site_packages中。自行百度解决or参照:https://blog.csdn.net/WHUGIS1805/article/details/129702200

TODO

1.收集相关图像数据训练更多类型胶片模拟 ❎

2.尝试统一所有设备(iphone/安卓/佳能/尼康等相机)到胶片模拟的映射 ✅

Update

  1. 【24.01.21|Beta1.0】
  • Velviannnlp等胶片滤镜6个
  • GUI界面
  1. 【24.02.05|Beta1.1】
  • 新增富士nccc两个胶片模拟
  • 新增保存路径提示弹窗
  • 修复png图像加载出错问题,图像格式为opencv所支持的图像类型
  • 修复图像加载显示问题,保留原始图像比例进行自适应填充
  1. 【24.06.14】
  • 新研究了一种统一的色彩模拟方案,包括去色上色两个阶段,详情参考idea/自适应图像色彩迁移方案.md
  1. 【24.07.4】
  • 新增provia滤镜,支持点击/拖拽上传图像,支持模型切换

About

generate any filter like fuji/ricoh/olympus/sony/canon/nikon freely by neural network!

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages