基于深度学习的老(旧)照片上色算法模型设计与实现(对抗生成式网络GAN、含Github代码与Web端设计)
计算机毕业设计--基于深度学习技术(Transformer、GAN)的图像修复算法(含Github代码+GUI与Web端展示界面)
更多基于深度学习的毕业设计请关注专栏 --- 计算机毕业设计
✨ 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”中检查不同步骤产生的修图结果。
下图展示划痕检测结果与修复结果
修复效果展示:
修复效果展示:
注: 由于该模型是用256*256个图像预训练的,因此使用该模型需要将图像分辨率(像素)调整为256。
本科毕业学生选择做深度学习相关的毕业设计,不可能完全专注于算法,因此本项目配套设计了一个基于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毕业设计,只要你想得出,没有做不出