Image Classification examples with Convolutional Neural Networks in Pytorch Framework
House image dataset link: https://drive.google.com/file/d/10pIGHCyPXR0a6XxORTf_hoeEQgAxBixw/view?usp=sharing
Vgg16_classify file has training a vgg16 model using transfer learning.
#Project Codes references are as follows : 1)https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html
2)https://pytorch.org/tutorials/beginner/finetuning_torchvision_models_tutorial.html