CNN Detection (CS550 Machine Learning Project)
Team Members (Tensor) :
- Yadava Kishore Chodipilli (11940310)
- Thashmitha BS (11941250)
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.)
This is a work done in our CS550 course. We took the model from 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)