Pytorch快速分类微型框架

介绍

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

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

框架介绍

本项目基于 PytorchModelTools 项目二次抽象重构。

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

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

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

环境

理论上什么环境都能使用，因为我的代码并没有用到什么太复杂的操作。

Pytorch 1.8.1
cuda 11.1
Python 3.6

需要安装的第三方包

pip install xlwt tqdm matplotlib numpy pandas

使用说明

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

(可选)在自己的电脑上持续使用

python setup.py install