【Atlas 200DK】YOLOv3和YOLOv5部署
数据集介绍
无论什么数据集开发版移植的流程是差不多的,我们使用的数据集是NWPU VHR-10 Dataset,这是由西工大标注的航天遥感目标检测数据集,该数据集有650张包含目标的图像和150张背景图像,共计800张,目标种类包括飞机、船舰、油罐、棒球场、网球场、篮球场、田径场、港口、桥梁和汽车一共10个类别,为了满足项目要求,将四个篮球场归为一类,所以本项目实际类别有7个类别。我们进行数据增强扩充后总共有2130张,训练集有1420张,下面统计了每一个类别的个数:
类别 |
飞机 |
船只 |
油库 |
操场 |
港口 |
桥梁 |
车辆 |
数量 |
3378 |
930 |
718 |
2418 |
798 |
374 |
2254 |
开发板环境搭建
我们所使用的版本是C73,这个版本是和20.1.rc1兼容的,华为推出了1.3.0.0、1.32.0.0、20.0.RC1、2