CNN Detection (CS550 Machine Learning Project)

Team Members (Tensor) :

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 :