使用只有模型权重而没有模型类的cleverhans

Question

我使用的是别人创建的预训练模型，他们只发布了模型权重。目前，我正在将模型权重导入到我的图中，并通过张量名称调用它们。然而，这似乎与cleverhans的代码不兼容，该代码似乎需要具有方法predict的模型对象。

有没有什么办法可以解决这个问题，因为我没有model类和预测方法，所以不需要我重写大多数cleverhans攻击？

Answer 1

您描述的内容应该是可能的，但可能会对资源造成一定程度的消耗，因为它可能会多次重新创建图形。基本上，您可以在init方法中实现一个接受图形检查点的CleverHans模型类。get_logits或fprop方法应该接受一个输入张量，并通过执行一些图操作来加载图以获得相应的输出张量，从而将检查点图的输入张量替换为您自己的张量:请参见`tf.import_graph_de：https://www.tensorflow.org/api_docs/python/tf/graph_util/import_graph_def中的input_map参数