This project organizes classic images classification neural networks based on convolution or attention, and writes training and inference python scripts

Arwin-Yu Last update: Feb 18, 2024

Deep Learning Image Classification Models Based CNN or Attention

This project organizes classic Neural Networks based convolution or attention mechanism, in order to solve image classification.

This readme is introduced in Chinese (including most of the comments in the code). Please translate it into English if necessary.

1. Introduction

本项目包含了自 AelxNet 以来经典的深度学习图像分类模型，大部分模型是基于卷积神经网络的，也有一部分是基于注意力机制的。博客链接是对模型的介绍，会持续更新...

在项目目录中，模型的搭建代码在classic_models文件夹中；所有的模型训练代码和推理代码都是共用的，只有模型搭建代码不同，训练代码有三个不同的版本：

train_sample.py是最简单的实现，必须掌握，以下两个版本看个人需求。
train.py是简单升级版的实现，具体改进的地方见train.py脚本中顶部的注释。
train_distrubuted.py支持多gpu分布式训练。

最后，test.py是推理脚本，用于使用训练好的模型。dataload中是数据集加载代码；utils是封装各种功能的包，包括学习策略，训练和验证，分布式初始化，可视化等等。建议先学习掌握classic_models，train_sample.py和test.py这三部分，其他部分用到的时候再学习。

2. Dataset And Project

本项目是使用python语言基于pytorch深度学习框架编写的。

默认的数据集是花朵数据集，此数据集包含五种不同种类的花朵图像，用于训练的图像有3306张，用于验证的图像有364张。下载链接如下：https://pan.baidu.com/s/1EhPMVLOQlLNN55ndrLbh4Q 提取码：7799 。

下载完成后，记得在训练和推理代码中，将数据集加载的路径修改成自己电脑中下载存储的路径。

数据集图像展示如下：

开启模型的训练只需要在IDE中执行train_sample.py脚本即可；或者在终端执行命令行python train_sample.py 训练的log打印示例如下：

将模型用于推理只需要在IDE中执行test.py脚本即可；或者在终端执行命令行python test.py 给一张向日葵的图像，模型的输出结果示例结果如下：：

3. Methods And Papers

以下是本项目支持的模型列表

1. AlexNet
  Blog Introduction Link: 博文传送门
1. ZFNet
  Blog Introduction Link: 博文传送门
1. VggNet
  Blog Introduction Link: 博文传送门
1. GoogleNet
  Blog Introduction Link: 博文传送门
1. ResNet
  Blog Introduction Link: 博文传送门
1. DenseNet
  Blog Introduction Link: 博文传送门
1. SENet
  Blog Introduction Link: 博文传送门
1. MobileNet
  Blog Introduction Link: 博文传送门
1. ShuffleNet
  Blog Introduction Link: 博文传送门
1. EfficientNet
  Blog Introduction Link: 博文传送门
1. RepVGG
  Blog Introduction Link: 博文传送门
1. Vision_Transformer
  Blog Introduction Link: 博文传送门
1. Swin_Transformer
  Blog Introduction Link: 博文传送门
1. Visual Attention Network Blog Introduction Link: 博文传送门
1. ConvNeXt Blog Introduction Link: 博文传送门
1. MLP-Mixer Blog Introduction Link: [博文传送门]
1. AS-MLP Blog Introduction Link: [博文传送门]
1. ConvMixer Blog Introduction Link: [博文传送门]
1. MetaFormer Blog Introduction Link: [博文传送门]

This project organizes classic images classification neural networks based on convolution or attention, and writes training and inference python scripts

Deep Learning Image Classification Models Based CNN or Attention

1. Introduction

2. Dataset And Project

3. Methods And Papers

DRF-like declarative viewsets for FastAPI

Python ticketing utility for working with tickets in popular tools

HUB REST API Python bindings

:evergreen_tree: Python module for communicating with the Taiga API

IDEX v3 Exchange REST API python implementation

Serving a keras model (neural networks) in a website with the python Django-REST framework.

Python library to interface with Gerrit's REST API

Code repository for Building RESTful Python Web Services, published by Packt

⛔️ [DEPRECATED] python wrapper for Firebase's REST API

Quickly turn command-line applications into RESTful webservices with a web-application front-end. You provide a specification of your command line application, its input, output and parameters, and CLAM wraps around your application to form a fully fledged RESTful webservice.