THUYimingLi / BackdoorBox

The open-sourced Python toolbox for backdoor attacks and defenses.
GNU General Public License v2.0
457 stars 73 forks source link

毒化数据的保存格式与位置 #59

Closed Magipoet closed 1 year ago

Magipoet commented 1 year ago

你好,在BackdoorBox-main/BackdoorBox-main/tests/test_BadNets.py中,毒化数据是如何保存的呢?因为在SCALE-UP-main/SCALE-UP-main/test_BadNets.py文件中保存的毒化样本似乎有着格式问题。

THUYimingLi commented 1 year ago

你好,在BackdoorBox-main/BackdoorBox-main/tests/test_BadNets.py中,毒化数据是如何保存的呢?因为在SCALE-UP-main/SCALE-UP-main/test_BadNets.py文件中保存的毒化样本似乎有着格式问题。

感谢对我们工作的关注!你可以参考这里面的代码(链接: https://pan.baidu.com/s/1LiGVjLGx11adv6G7P6Y5nw 提取码: qijt)。首先执行对应攻击的test代码存下毒化数据,然后运行dataloader2tensor.py得到SCALE-UP需要的数据类型。

Magipoet commented 1 year ago

这个文件一直下载失败,请问里面的代码在github上有保存吗,如果没有的话,可否麻烦你发送到邮箱1033622396@qq.com。 在SCALE-UP-main/test_BadNets.py文件中,有着良性样本的保存“torch.save(test_samples, 'benign_test_samples.pth')”,而毒化样本的保存格式“torch.save(poisoned_test_dataset, 'poisoned_test_dataset_BadNets.pth')”与良性样本是否一致呢?

THUYimingLi commented 1 year ago

这个文件一直下载失败,请问里面的代码在github上有保存吗,如果没有的话,可否麻烦你发送到邮箱1033622396@qq.com。 在SCALE-UP-main/test_BadNets.py文件中,有着良性样本的保存“torch.save(test_samples, 'benign_test_samples.pth')”,而毒化样本的保存格式“torch.save(poisoned_test_dataset, 'poisoned_test_dataset_BadNets.pth')”与良性样本是否一致呢?

这部分代码github里没有,或者说没有纯净版的。我刚在我这边本地测试过,可以下载,不过还是邮箱发你一份了。具体的保存格式是一样的,但是你应该还需要一个转换格式的代码(dataloader2tensor.py)才能适配SCALE-UP(具体见我发你的代码示例)。

Magipoet commented 1 year ago

好的,非常感谢!