/NN_Demo

用个小Demo用来记录NN学习,包括数据处理,训练和部署等全流程

Primary LanguagePython

本项目用来以最简单的demo的形式展示神经网络的应用过程,用来串联其中需要用到的技术

  • 1.数据准备
  • 2.模型训练
  • 3.模型部署

依赖环境: opencv pytorch

执行顺序:

  • 1.执行DataCollection中的mnist2image.py文件,会转换原始数据为训练测试要用到的jpg和txt
  • 2.执行Training中的train.py文件,完成模型训练,并保存当前模型和训练过程
  • 3.执行Development中的use_demo.py可以转换你自己手写的数字为模型推理结果

参考文献: