一、项目出发点
AI Studio为我们提供了免费的GPU资源,当我们在NoteBook环境中把代码调试成功后,通常一个训练任务耗时较长,而Notebook离线运行有时长限制,一不小心就容易被kill掉。
如何解决这一问题?
后台任务帮到你!有关什么是后台任务,以及如何起一个后台任务,官方已经出了相关教程。
本次分享将基于笔者的一个任务需求-SAR图像目标检测,带领大家从0到1跑通一个检测任务的后台训练,希望能为有类似需求的同学提供一点帮助。
二、Notebook离线调试
2.1 数据集准备
这一步是为模型训练做好准备
数据集获取有两种方式:
- 在AI Studio平台-数据集中搜索是否有自己需要的数据集;
- 如果是自己收集的数据,首先需要制作成为VOC 或者 COCO 格式的数据,这里笔者已经把一个SAR图像目标检测数据集SSDD制作好了,然后上传到了AI Studio平台SSDD遥感SAR目标检测数据集-COCO格式
2.2 新建Notebook任务
新建一个Notebook任务, AI Studio平台注册账号后,点击创建项目-选择NoteBook任务,然后添加上一步的数据集,参考下图操作(注意数据集选用SSDD遥感SAR目标检测数据集-COCO格式),完成项目创建。
2.3 环境准备
本次任务