PaddleColorization-黑白照片着色
将黑白照片着色是不是一件神奇的事情/p>
本项目将带领你一步一步学习将黑白图片甚至黑白影片的彩色化
黑白照片着色
我们都知道,有很多经典的老照片,受限于那个时代的技术,只能以黑白的形式传世。尽管黑白照片别有一番风味,但是彩色照片有时候能给人更强的代入感。本项目通过通俗易懂的方式简单实现黑白照片着色并对部分照片取得不错的着色效果。黑白照片着色是计算机视觉领域经典的问题。近年来随着卷积神经 络(CNN)的广泛应用,通过CNN为黑白照片着色成为新颖且可行的方向。本项目承载于百度的学习与实训 区AIStudio,整体实现采用ResNet残差 络为主干 络并设计复合损失函数进行 络训练。
开启着色之旅!!!
先来看看成品
2.使用Shell命令对数据集进行初步处理(运行时间:约20min)
mkdir: cannot create directory ‘work/train’: File exists
mkdir: cannot create directory ‘work/test’: File exists
3.预处理
3.1预处理-采用多线程对训练集中单通道图删除(运行时间:约20min)
done!
已删除数量:470
done!
已删除数量:432
done!
已删除数量:426
done!
已删除数量:483
[Errno 2] No such file or directory: ‘./work/train/n02105855_2933.JPEG’
done!
已删除数量:490
done!
已删除数量:454
done!
已删除数量:467
done!
已删除数量:482
3.2预处理-采用多线程对图片进行缩放后裁切到256*256分辨率(运行时间:约40min)
from PIL import Image import os.path import os import threading from PIL import ImageFile ImageFile.LOAD_TRUNCATED_IMAGES = True '''多线程将图片缩放后再裁切到256*256分辨率''' w = 256 h = 256 def cutArray(l, num): avg 声明:本站部分文章及图片源自用户投稿,如本站任何资料有侵权请您尽早请联系jinwei@zod.com.cn进行处理,非常感谢!