需求:
训练一个模型,可以判断一张图片上有无水印。
方法:
水印有小水印和满屏水印2种。
方法1:
可以作为分类任务,使用深度学习模型。
方法2:
可以作为目标监测,把主图当成背景文本,把水印当作检测的目标。
实际做下来目标检测的效果会更好一点。
数据集
数据集如果够大,可以让有水印和无水印的数据集不是基于同一张图。
数据集切分:如果数据集1w以上,可以使用8:1:1,或者7:2:1.
训练收敛:看数据集大小,快的可能10个epoch,慢的可能要上百个epoch。
大模型
chat gpt 准确率0.8
通义千问 准确率0.7
百灵 准确率0.42
自己的目标检测代码0.76