physicube / CSED499_Project

CSED499_Project

Directory Structure

• data : directory for image datas.
• data_loader : codes for loading images from data directory
• docs : document directory
• model : codes for pytorch models and loading weights
• trainer

Train models

• python train_face.py
• python train_flower.py
• python train_gtsrb.py model weights are saved in saved/ directory

Attack Image generation

• python train_*.py : train model using transfer leaning
• python train_attack_*.py : train attack image using student models generated by transfer learning.