【百度大脑AI计算盒FZ5C】用 AI Studio 训练物体检测模型

robe.zhang

【百度大脑AI计算盒FZ5C】用 AI Studio 训练物体检测模型

AI Studio 是百度 AI 开发环境，在云端，连着线就可以用，已经部署好的 paddlepaddle 开发环境，本文就使用AI 开发环境，使用ShuffleNetV2 网络，训练一个物体检测模型。

这个是AI Studio 用户中心：https://aistudio.baidu.com/aistudio/usercenter

最近创建的项目：Edgeboard 高性能目标检测，网络是 ShuffleNetV2

关于 ShuffleNetV2 网络的介绍：

点一下启动项目，再点进入项目：

选择高级开发环境：

这个网络需要在 GPU 上跑，CPU 不行。高级开发环境配置很不错，比当前个人最新的主机都好，GPU 是英伟达 Tesla V100 显卡，显存 16GB，CPU 四核Intel(R) Xeon(R) Gold 6148 CPU @ 2.40GHz，32GB内存，100GB硬盘，笔者的体验，好的很，哈哈哈，真的很不错

启动成功，进入开发环境：

Notebook 界面：

终端界面：

还有一个AI studio 图形界面开发，参考笔者另一篇文章：

【百度大脑AI计算盒FZ5C】百度 AI Studio 图形界面开发

https://www.cirmall.com/bbs/thread-204468-1-1.html

本次训练全部在 notebook 界面完成，终端界面和 notebook 界面差不多，看看就会了，不介绍

解压数据集：数据是借用别人的，2.8G的数据，标记好的

开始训练：需要用 GPU 环境：

如果没有 GPU 环境，使用 CPU 环境应该也行，自己修改一下参数，估计训练时间会很长吧：

运行过程能看到系统资源使用情况：

训练完成固化模型：

预测，可视化，可以正常检测得到人

转换模型：

打包模型：

点下载，保存模型，保存到电脑上：

AI Studio 训练完成后，要把项目停掉，不然会继续运行，会消耗GPU资源

AI Studio 训练模型完成了