This is the code repository for Evolutionary Trigger Set Generation for DNN Black-Box Watermarking and Watermarking Deep Neural Networks for Embedded Systems.

The models come from https://github.com/kuangliu/pytorch-cifar.

To run watermarking:
python watermarking_main.py

To run evolution:
python evolution_main.py 0.2