Skip to content

在图像的采集和传输过程中,常常会产生各种形式的损坏,这些损坏会降低对图像信息的准确解释,同时也有一些老照片因保存不当而出现污渍或者划痕缺失的情况。本课题结合变分编码器(VAEs)、Vision Transformer与对抗生成网络(GAN)的深度学习算法和图像处理技术为基础,设计开发了一款图像修复深度学习算法小程序。这款程序为用户带来了基于 BS 结构的 Web 应用程序。用户操作时,只需轻松选定待修复的照片即可还原一张真实图像。

Notifications You must be signed in to change notification settings

zxx1218/oldImage_fix

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

基于深度学习的老照片修复(清晰化+划痕修复+色彩增强)算法设计与实现

本篇文章是针对老(旧)照片色彩复原、上色,如果您想对灰白老照片进行上色,色彩复原,请参考这篇CSDN作品👇

基于深度学习的老(旧)照片上色算法模型设计与实现(对抗生成式网络GAN、含Github代码与Web端设计)

本篇文章是针对老(旧)照片色彩复原、上色,如果你想对有大面积破损的图像进行修复,请参考这篇CSDN作品👇

计算机毕业设计--基于深度学习技术(Transformer、GAN)的图像修复算法(含Github代码+GUI与Web端展示界面)

更多基于深度学习的毕业设计请关注专栏 --- 计算机毕业设计

✨ Demo_1

✨ Demo_2

✨ 图像修复在线体验

👇

Web端在线体验地址:访问这里进行老照片修复在线体验

在线体验使用方式:

  打开连接后,左侧两个图片输入框分别需要上传待修复的原图。在网页下方提供了若干组输入样例,点击样例自动填充到相应位置后即可点击开始修复查看效果。

注意: 由于服务器配置较低,因此修复过程比较长,请您在点击“开始修复”按钮后多等待一会儿(大约40s-60s)

运行要求

该代码需要在安装了Nvidia GPU和CUDA的计算机上运行,运行代码需要Python>=3.9。


clone项目后,只需一个简单的命令即可轻松恢复旧照片。 对于没有划痕的图像:
python run.py --input_folder [test_image_folder_path] \
              --output_folder [output_path] \
              --GPU 0

对于带有划痕图像:

python run.py --input_folder [test_image_folder_path] \
              --output_folder [output_path] \
              --GPU 0 \
              --with_scratch

对于带有划痕的高分辨率(就是说图像清晰度很高)图像

python run.py --input_folder [test_image_folder_path] \
              --output_folder [output_path] \
              --GPU 0 \
              --with_scratch \
              --HR

注意:请修改上述步骤中[ ]内的路径,模型修复完成后,可以在“output_path”中检查不同步骤产生的修图结果。

2) 2025新开发功能---照片划痕检测

下图展示划痕检测结果与修复结果

3) 照片全图结构恢复(对老照片结构翻新,相当于清晰化,结构化)

修复效果展示:

4) 面部增强

修复效果展示:

: 由于该模型是用256*256个图像预训练的,因此使用该模型需要将图像分辨率(像素)调整为256。

5) 一个用户友好的GUI界面

本科毕业学生选择做深度学习相关的毕业设计,不可能完全专注于算法,因此本项目配套设计了一个基于pyqt(python提供的一个图形化界面)的GUI界面,它接受用户待修复的照片图像输入,并在相应的窗口中显示结果。

如何使用?

1.在pycharm中运行GUI.py文件。

2.单击浏览并选择待修复的图像以修复划痕(test_images/old_w_scratch文件夹中提供了可用于测试的图像)。

3.单击“修改照片”按钮。

4.等待一段时间(控制台会有算法的中间过程输出),然后在GUI窗口上查看修复结果。

5.单击“保存”,或者“退出窗口”退出,保存后的图片会在指定的文件夹中获取到。

如何自行训练模型?

  • 训练自己想要的定制化图像修复模型只需要准备好针对需要修复图片的同类型图片数据即可,具体训练方式请咨询作者

作者联系方式:

  • VX:Accddvva
  • QQ:1144968929
  • Github提供训练好的模型文件以及调用该文件进行修图的测试代码(Github开源代码中提供可供测试的模型文件,拉取Git代码后安装环境可进行修图测试,但不含模型定义源码、训练代码以及桌面端GUI代码,
  • 该项目完整代码 + 训练好的模型文件 + 详细环境配置(如有需要可提供模型远程部署)+ GUI界面 + Web端 + 训练方式 == 价格300RMB
  • 如果你的电脑没有合适的GPU,或需要自行训练私有数据而自己的GPU现存或性能不够,作者另提供GPU服务器短期租赁服务,24G显存+64G内存云服务器每月 50RMB

广告

  • 作者于浙江某985高校就读人工智能方向研究生(CSDN已认证),可以定制模型,并提供相应技术文档以及各种需要,只需要描述需求即可
  • 人工智能、深度学习领域,尤其是计算机视觉(Computer vision,CV)方向的模型or毕业设计,只要你想得出,没有做不出

About

在图像的采集和传输过程中,常常会产生各种形式的损坏,这些损坏会降低对图像信息的准确解释,同时也有一些老照片因保存不当而出现污渍或者划痕缺失的情况。本课题结合变分编码器(VAEs)、Vision Transformer与对抗生成网络(GAN)的深度学习算法和图像处理技术为基础,设计开发了一款图像修复深度学习算法小程序。这款程序为用户带来了基于 BS 结构的 Web 应用程序。用户操作时,只需轻松选定待修复的照片即可还原一张真实图像。

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages