# Pytorch快速分类微型框架

**Repository Path**: l1binn/fast_classification_framework

## Basic Information

- **Project Name**: Pytorch快速分类微型框架
- **Description**: 基于Pytorch开发的快速分类模型，易于修改、易于使用。
- **Primary Language**: Python
- **License**: MulanPSL-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 1
- **Created**: 2021-06-18
- **Last Updated**: 2024-04-03

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# Pytorch快速分类微型框架

### 介绍
基于Pytorch开发的中文快速分类框架，易于修改、易于使用。

**框架用途：** `图像分类`、`模型训练`

#### 框架介绍
本项目基于 [PytorchModelTools](https://gitee.com/lb1098/PytorchModelTools) 项目二次抽象重构。

- 之前的代码太过于糟糕(**可移植性差**)，虽然能用，但是可移植性差，代码耦合高。

因此本项目会更加精简、好用、可移植性强，且全中文，可能会用到一些python高级语法。

**（敬请期待）后期会提供在shell页面进行训练**

#### 环境
理论上什么环境都能使用，因为我的代码并没有用到什么太复杂的操作。
```
Pytorch 1.8.1
cuda 11.1
Python 3.6
```

#### 需要安装的第三方包
```shell
pip install xlwt tqdm matplotlib numpy pandas
```

### 使用说明
框架功能已经和原框架一致，可通过`demo/lenet5_cifar10.py`和`demo/resnet_gemstones.py`快速了解。

### (可选)在自己的电脑上持续使用
```shell
python setup.py install
```