从数据采集到部署，手把手带你训练一个高质量的图像分类模型

文章预览

关注 「 AIWalker 」 并 星标 从此AI不迷路 来源于 OpenMMLab ，作者 李剑锋 MMPreTrain 是一款基于 PyTorch 的开源深度学习预训练工具箱，本文将从数据采集到部署，手把手带大家使用 MMPreTrain 算法库训练一个高质量的图像分类模型。 MMPreTrain 项目链接： https://github.com/open-mmlab/mmpretrain （文末点击阅读原文可直达， 觉得好用欢迎点亮小星星） 1.环境配置 对于MMPreTrain的环境配置我们可以通过官方的教程来实现。 首先，我们要创建一个MMPreTrain的虚拟环境并安装PyTorch。安装GPU版本我们可以选择CPU或者GPU进行预测，而安装CPU版本就只能使用CPU进行预测。而我之所以采用下面这个方式是因为有些时候通过官方的这个代码安装GPU版本，但是实际安装的是CPU版本，因此我便在官网选择了pip的方式进行安装。假如CUDA版本比较低的话，也可以到官网查找到对应的下载代码 ………………………………