Skip to content

基于MindYOLO的模型训练、推理与加速 #21

@SilvaXiao

Description

@SilvaXiao

请下载链接里的参考文件, 提取码: SJTU。:
https://pan.sjtu.edu.cn/web/share/335e16661e50a60cf5afca7032be717a
说明:
YOLO-TrainTest.ipynb:是我们利用Ultralytics工具(基于Torch)开发的YOLO训练+推理的脚本。可以自动下载coco128数据集进行训练,也支持换自己标注好的数据集(只需要改写一个yaml文件并替换就可以用自己的数据集进行训练),很方便用于教学案例。
Drink_284_Detection_Labelme:文件夹下是我们自己标的饮料数据集

需求:
1、是否可以替换成基于MindYOLO的训练与推理方案(可以不限定mindspore版本)?并且,需要在华为云和本地都能成功运行,请标注好具体需要的云上镜像和本地mindspore配套环境哈。云上可以测试一张照片进行推理。
2、需要转换成MINDIR进行加速,因为之前测试下来感觉速度很慢,但张照片都需要推理很久。
3、如果可以,请部署到香橙派开发板进行推理测试。

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions