Skip to content

Implementation of some popular CNNs (VGG-Net, Res-Net, Mobile-Net) for image classification on CIFAR-10 dataset with PyTorch library

Notifications You must be signed in to change notification settings

parham1998/CNNs-PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CNNs-PyTorch

Implementation of some popular CNNs (VGG-Net, Res-Net, Mobile-Net) for image classification on CIFAR-10 dataset with PyTorch library

references

K. Simonyan, and A. Zisserman.
"Very deep convolutional networks for large-scale image recognition" (ICLR - 2015)

K. He X. Zhang S. Ren, and J. Sun.
"Deep residual learning for image recognition" (CVPR - 2016)

A. G. Howard, M. Zhu, B. Chen, D. Kalenichenko, W. Wang, T. Weyand, M. Andreetto, and H. Adam.
"Mobilenets: Efficient convolutional neural networks for mobile vision applications" (arXiv - 2017)

M. Sandler, A. Howard, M. Zhu, A. Zhmoginov, and L.-C. Chen.
"MobileNetV2: Inverted Residuals and Linear Bottlenecks" (CVPR - 2018)