摘要: TensorFlow Lite+OpenCV实现移动端水印的检测与去除 闲鱼技术:镇雷 概要: 本篇文章介绍了TensorFlow Lite与OpenCV配合使用的一个应用场景,并详细介绍了其中用到的SSD模型从训练到端上使用的整个链路流程。
TensorFlow Lite+OpenCV实现移动端水印的检测与去除
概要:
本篇文章介绍了TensorFlow Lite与OpenCV配合使用的一个应用场景,并详细介绍了其中用到的SSD模型从训练到端上使用的整个链路流程。在APP中的使用场景为,用户在发布图片时,在端上实现水印的检测和定位,并提供去水印的功能。
具体步骤有:
1,使用TensorFlow Object Detection API进行SSD模型的训练
2,模型的优化和转换,模型在端上的解析使用(本篇主要使用iOS端的C++代码作为示例)
3,将输出locations值通过NMS(非极大值抑制)算法得到最优的框
4,使用OpenCV去除水印
使用的库及工具:
1、TensorFlow v:1.8r +
2、TensorFlowLite v:0.0.2 +
3、OpenCV
4、labelImg
SSD检测并定位水印
SSD简介
SSD,全称Single Shot MultiBox Detector,是Wei Liu在ECCV 2016上提出的一种目标检测算法,截至目前是主要的检测框架之一,