基于fastai的分类网络

元气小坏坏 提交于 2020-01-20 12:23:38

classify.fastai

Classifier based on fastai

欢迎访问:https://github.com/pprp/classify.fastai/

字符识别分类器-基于fastai进行构造

以往的fastai教程都是很简单的几行进行一个模型的训练,对初学者来说看起来很简单。但是仅仅训练一个分类的网络是远远不够的,我们还需要各种指标对分类的效果进行评判。

这个库集中了目前比较好的几篇中文博客,并且在此基础上查阅了fastai的API教程(😲但是不得不说 感觉做的没有keras对于tensorflow那样好用,另外用的人感觉不是很多。)进行综合,主要是基于jupyter notebook进行编辑的,py文件也有,但是还不完善,欢迎pull request​ 😄

环境配置

建议使用anaconda3进行环境的配置和管理,目前使用的fastai以及pytorch都是最新版本的。链接

  • pip安装

    •    pip install fastai
         pip install jupyter notebook jupyter_contrib_nbextensions
  • conda安装

    •   conda install jupyter notebook
        conda install -c conda-forge jupyter_contrib_nbextensions
  • 之后启动命令jupyter notebook 然后在网页中打开并运行即可。

主要功能

  • 提供了数据,在data文件夹下,解压即可使用
    • 如果需要跑自己的分类器,直接替换train,valid等文件夹下的内容即可。
    • 格式为:每个类别一个文件夹,文件夹内放置图片。对命名没有要求。
  • 查找合适的learning rate
    • 该功能主要由fastai提供
    • 这个最低点就是合适的loss取值。(ps:有时候感觉不准)
  • 训练模型只需要通过参数传递即可
    • vgg16_bn,vgg19_bn,alexnet
    • resnet18,resnet34,resnet50,resnet101,resnet152
    • squeezenet1_0,squeezenet1_1
    • densenet121,densenet169,densenet201,densenet161
  • 训练过程中实时画train-valid-loss图(jupyter notebook很强大)
    • 像这种图就属于比较完美的,证明learning rate以及数据集安排都很合适。
    • 此外还会展示实时loss-error_rate-accracy等
  • 完成训练后loss可视化
  • 训练过程learning rate以及momentum可视化
  • 查看主要有哪些图像画错了top error
  • 查看混淆矩阵
  • 查看所有类别
  • 将数据集部分内容可视化

说明😎

其中可能有很多内容无法运行,由于本人也是初学者,仅通过看博客与API进行fastai的学习,朋友们有时间可以看fastai提供的课程,相信在里边会有详细的讲解。:hugs:

目前主要内容集中于train_notebook.ipynb文件中欢迎大家进行fork,star,pull requests。

待完成工作:thinking:

  • 训练模型保存。
  • 模型结构的设计与改动(尝试了一点但是没有解决bug)。
  • 训练模型的加载与预测。
  • 更好的可视化支持。
  • 更多的分类性能指标。
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!