/CV_ImageClassification

Code for ImageClassification based on Pytorch

Primary LanguageJupyter Notebook

Classification

This repository mainly includes Code for image classification 这个库的内容是基于Pytorch框架实现的图像分类代码

  1. AlexNet
  2. VGG
  3. ResNet (include Saliency Map)