原创

: CAT-Net: Compression Artifact Tracing Network 算法测试

CAT-Net: Compression Artifact Tracing Network 算法测试

2021年的开源图像篡改检测算法还是很难找的

这篇文章是 WACV2021 paper,作者是韩国人。

WACV据说是仅次于CVPR等的顶会。

https://github.com/mjkwon2021/CAT-Net
这个东西直接在github 上面搜索名字竟然搜不到。
但是会出现readme文档,在这个readme文档中点进去才会出现代码。

算法训练使用的数据集

最好不要使用下面的数据集进行测试
在这里插入图片描述
在这里插入图片描述

对于一个培训集 使用以下数据集
auth.表示未篡改图像;tamp.表示篡改图像

安装库版本

这些在requirements文件里面都有
torch==1.1.0
Cython
pyyaml
yacs>=0.1.5
tqdm
torch_dct
pillow
opencv-python
tensorboardX
matplotlib
seaborn

权重文件

权重文件需要科学上网
Download weights from link.

link:https://drive.google.com/drive/folders/1hBEfnFtGG6q_srBHVEmbF3fTq0IhP8jq?usp=sharing
是谷歌的下载地址。

百度云的下载地址在这里,模型很大,800M.

链接:https://pan.baidu.com/s/10hjC7u_lL8TX7NfPtioFzQ
提取码:4tlj

测试

在casia 数据集中,是非常惊艳的效果了。不如说这个效果过于惊艳了,和gt标注的完全相同。

但是也有在训练集中过拟合的嫌疑。

casia V2

请添加图片描述
请添加图片描述

请添加图片描述
请添加图片描述

coverage

请添加图片描述
请添加图片描述

请添加图片描述
请添加图片描述

在这里插入图片描述
COVERAGE数据集中的效果不是很好。说明此算法的泛化能力不好。在coverage的多数图像中都未能达到训练集(casia)中的效果。

有一点要注意

若输出为热力图,则输出为固定的某一大小。

若将灰度图保存出来,则分辨率为原图像的1/4,长和宽都是1/4.

计算指标时候,需要进行resize,可能会有一些指标计算的损失,不过对于区域检测而言,几个像素的误差根本就不是问题。

抗干扰性能

CASIA数据集某图像。
应该是算法的一张训练图像。

发现,在旋转某些角度时,检测结果几乎完美,但是,但旋转超过180度,检测结果将会大打折扣。
不明原因。原则上,DCT矩阵不会被旋转破坏的。
在这里插入图片描述

请添加图片描述
请添加图片描述