如何证明您的模型属于您:基于盲水印的框架来保护DNN的知识产权
这是一个示例示例(在MNIST和CIFAR-10上),该示例演示了如何运行介绍的框架
快速开始
配置环境
Python=3.7
Pytorch=1.2.0
Others are the latest version
数据集
更改args.dataset='mnist or cifar10'定义数据集。 数据集将自动下载到/data 。
如果args.dataset='mnist' ,则排他徽标是从mnist数据集中随机抽取的样本。
如果args.dataset='cifar10' ,则专用徽标位于/data/IEEE/logo 。 您还可以选择其他徽标。
主机模型
如果args.dataset='mnist' ,则只能实现将水印嵌入lenet1, lenet3, lenet5 。
如果args.dataset='cif
2021-10-11 21:05:41
2.45MB
Python
1