CNN Detection (CS550 Machine Learning Project)

Team Members (Tensor) :

1. Yadava Kishore Chodipilli (11940310)
2. Thashmitha BS (11941250)

This is a work done in our CS550 course. We took the model from this github link for learning. We retrained it with a different dataset (more details can be found inside CNN Detection folder's README file and our project report pdf.)

Our Work : Original model was trained on progan dataset with a resnet50 classifier and we trained our model with stylegan dataset (relatively very small dataset, obtained from original model's testing dataset) with resnet101 and resnet152 classifiers. Results and trained models are provided for your reference, links can be found inside main code folder's README file.

Folder Structure :

• CNN Detection (code folder)
• Demo_video (small demo of the code, running in google colab)
• Project Report (report of our project)
• Project Presentation (presentation slides of our project)