Blind-Watermark-for-DNN:“如何证明模型属于您”论文的实施-源码

如何证明您的模型属于您：基于盲水印的框架来保护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