手写文字擦除训练数据生成Mask的方法
收藏
比赛训练数据只提供了源数据和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
收藏
请登录后评论
这个内容非常棒,很有参考价值,非常感谢!!!
感觉这是一个很好的思路呀。在很多图片处理的时候都可以借鉴,谢谢大佬分享。
这个内容非常棒,很有参考价值,非常感谢!!!
大佬666,这就去试试
感谢分享,很好的内容