手写文字擦除训练数据生成Mask的方法
收藏
快速回复
百度网盘AI大赛——图像处理挑战赛 其他学习资料开源方案 7610 5
手写文字擦除训练数据生成Mask的方法
收藏
快速回复
百度网盘AI大赛——图像处理挑战赛 其他学习资料开源方案 7610 5

比赛训练数据只提供了源数据和GT,没有相应的Mask图片,可通过源数据和GT之间的diff来生成Mask文件,这里分享一个方法

# src_image为RGB三通道源图片,gt_image为RGB三通道GT图片
# threshold = 25
diff_image = np.abs(src_image.astype(np.float32) - gt_image.astype(np.float32))
mean_image = np.mean(diff_image, axis=-1)
mask = np.greater(mean_image, threshold).astype(np.uint8)
mask = (1 - mask) * 255

效果展示:源图像,GT,Mask

上述方法来自EnsNet作者的分享:https://github.com/HCIILAB/Scene-Text-Removal/issues/9#issuecomment-558943078

3
收藏
回复
全部评论(5)
时间顺序
三岁
#2 回复于2021-12

这个内容非常棒,很有参考价值,非常感谢!!!

0
回复
DeepGeGe
#3 回复于2021-12

感觉这是一个很好的思路呀。在很多图片处理的时候都可以借鉴,谢谢大佬分享。

0
回复
帝王爷
#4 回复于2021-12

这个内容非常棒,很有参考价值,非常感谢!!!

0
回复
嘟嘟
#5 回复于2021-12

大佬666,这就去试试

0
回复
霄勇善战
#6 回复于2021-12

感谢分享,很好的内容

0
回复
在@后输入用户全名并按空格结束,可艾特全站任一用户