Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
kenan976431 45d518a7e2 | 1 year ago | |
---|---|---|
poisons | 1 year ago | |
tabor | 1 year ago | |
README.md | 1 year ago | |
download_data.sh | 1 year ago |
This repository contains partial code implementation of the paper (https://arxiv.org/pdf/1908.01763.pdf). Currently this repo has been written to work on the GTSRB dataset with the 6 Conv + 2 MaxPooling CNN from the original paper.
This codebase is written in tensorflow and tf.keras and has been tested on tensorflow 1.14 and python 3.6.8
Clone the TABOR repository
git clone https://github.com/UsmannK/TABOR.git
Download the training data (GTSRB signs)
cd TABOR
./download_data.sh
Run the BadNet trainer:
python3 tabor/train_badnet.py --train --poison-type FF --poison-loc TL --poison-size 8 --epochs 10 --display
Currently supported options:
poison type: FF
(firefox logo) and whitesquare
poison location: TL
and BR
: Top Left and Bottom Right
poison size: integers
To train without any poison, exclude the poison-*
options
Run TABOR on the best model in output/
python3 tabor/snooper.py --checkpoint output/badnet-FF-10-0.97.hdf5`
The final mask and pattern will be written to mask.py
and pattern.py
TODO! Currently running TABOR for 500 epochs as in the paper takes about 4 hours on my hardware so generating the full results table is still upcoming.
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》