TensorFlow 生成对抗样本(未完待续)
2019-02-26 本文已影响7人
公输睚信
前面一系列文章已经全面的讲解了怎么使用 TensorFlow 来训练、预训练、保存、测试分类模型等内容,本文陈述怎么使用 TensorFlow 来简单的生成对抗样本攻击训练好的模型。
本文的实验数据来源于 kaggle比赛猫狗数据集百度网盘分享。这个数据集包含训练样本 25000 张,其中猫、狗以接近 1:1 的比例分布。我们先使用预训练模型 ResNet-50 来训练一个准确率较高的猫狗分类模型,然后使用该模型来生成对抗样本,让猫狗模型将本是 猫 的图片预测为 狗、本是 狗 的图片预测为 猫。
所有代码见 [GitHub]。