Skip to content

liudan193/poison_froggo

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Poison_Froggo

re-implementation of the poison frogs! paper in PyTorch by Shafahi et al.:

Run the attack

Create test and train folder as subfolders under ./poison_frog/datasets/ and place dog and fish named folders with train and test data inside. The original paper uses around 900 training images each and 698 test images for dog class and 401 test images for fish class. The imagenet data can be found here.

After preparing the data simply run

python run_frog.py

for a single attack with only one poison, or

python run_frog.py --num_poisons X --num_runs Y

for X poisons and Y attack runs. The attack will be evaluated when finished and will print the successrate depending on the total attack runs.

About

fixed the memory leaks problem

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%