mindspore是华为开源的深度学习框架,之前已经参加过两期训练营,8月8日又举办了第三期,继续参加学习。
这次主要的作业是用mindspore实现yolo_v3,并在华为云的modelArts平台上实现。
mindspore跑yolo_v3,其实在mindspore主仓的model_zoo里已经有了训练和测试的代码,可以到 官方gitee仓 里查看
但是官方的代码是在本地跑的,要在modelArts上跑,需要做一些适当的调整
首先,要把代码和数据集上传到云上,要在华为云上创建一个“对象储存服务obs”,将代码、数据集上传
其次,要创建modelArts训练作业
点击之后出现如下配置界面
最重要的两个参数,一个是数据集目录,一个是模型保存位置,默认以data_url
和train_url
为参数名通过argparse传入启动程序,所以启动程序中一定要有解析这两个参数的功能,并将涉及数据集的目录用data_url
替代,涉及模型保存的目录,用train_url
代替。
最后,上传数据的“对象储存服务”相当于云盘&