In this project, using VGG-16 to conduct transfer learning. The val set is used to verify whether my model works well. Eventually getting a CNN (Convolutional Neural Network) that can identify cat and dog pictures.
mmbell0525 / dog_vs_cat Goto Github PK
View Code? Open in Web Editor NEWThis project forked from xiutangzju/dog_vs_cat
Using pytorch and transfer learning to achieve a cat and dog classifier.