181 Star 1.1K Fork 271

MindSpore/community

【课程作业】基于昇思MindSpore实现YOLOv1

TODO
Task-Tracking
创建于  
2022-06-10 15:03

【背景描述】
YOLO之前的物体检测方法,如R-CNN,Fast-R-CNN,Faster-R-CNN等,主要是通过候选区域产生大量的可能包含待检测物体的预训练框,再用分类器去判断每个框里是否包含有物体,以及物体所属类别的概率或者置信度。YOLO不同于这些物体检测方法,它将物体检测任务当做一个回归问题来处理,使用一个神经网络,直接从一整张图像来预测出框的坐标、框中包含物体的置信度和概率。因为YOLO的物体检测流程是在一个神经网络里完成的,所以可以端对端的来优化物体检测性能。

【命题内容】

  1. 基于MindSpore Vision套件实现YOLO网络模型。

【数据集】
PASCAL VOC数据集(使用VOC2012的训练/验证集和VOC2007的测试集,镜像下载地址https://pjreddie.com/projects/pascal-voc-dataset-mirror/)

【参考内容】

  1. 论文地址:https://arxiv.org/pdf/1506.02640.pdf
  2. 参考代码地址:https://github.com/abeardear/pytorch-YOLO-v1

【产出标准】

  1. 完成YOLOv1的backbone和model接口的编写;
  2. 完成YoLOv1模型训练,验证,推理代码的编写,并能够成功跑通;
  3. 输出精度达标的预训练模型YOLO和YOLO VGG16。
    输入图片说明

【PR提交地址】
https://gitee.com/mindspore/vision

【答题要求】

  1. 必须使用昇思MindSpore框架,不得引入Pytroch和Tensorflow等第三方AI框架及其库。
  2. 作业提交内容包括精度Benchmark等,格式参考examples/classification/mobilenetv2。
  3. 提供该网络模型原论文或者地址提供的所有结构,避免只实现一种结构,并尽可能服用vision套件的API接口。
  4. 代码按照开源的方式上传https://gitee.com/mindspore/vision,边开发边上传,每个PR代码按规则不超过300行。
  5. 基于vision套件实现的网络模型后,提供算法原理+API代码+核心伪代码的ipython案例,详细参考resnet50。

评论 (2)

hujingsong 创建了Task-Tracking 3年前

Please assign maintainer to check this issue.
请为此issue分配处理人。
@hu-jingsong

Please add labels (comp or sig), also you can visit https://gitee.com/mindspore/community/blob/master/sigs/dx/docs/labels.md to find more.
为了让代码尽快被审核,请您为Pull Request打上 组件(comp)或兴趣组(sig) 标签,打上标签的PR可直接推送给责任人进行审核。
更多的标签可以查看https://gitee.com/mindspore/community/blob/master/sigs/dx/docs/labels.md
以组件相关代码提交为例,如果你提交的是data组件代码,你可以这样评论:
//comp/data
当然你也可以邀请data SIG组来审核代码,可以这样写:
//sig/data
另外你还可以给这个PR标记类型,例如是bugfix或者是特性需求:
//kind/bug or //kind/feature
恭喜你,你已经学会了使用命令来打标签,接下来就在下面的评论里打上标签吧!

hujingsong 修改了描述 3年前
hujingsong 修改了描述 3年前
hujingsong 修改了描述 3年前
hujingsong 修改了描述 3年前
fangwenyi 任务状态TODO 修改为ACCEPTED 3年前
fangwenyi 负责人设置为hujingsong 3年前
fangwenyi 关联项目设置为MindSpore Iteration Management 3年前
hujingsong 负责人hujingsong 修改为未设置 3年前
hujingsong 任务状态ACCEPTED 修改为TODO 3年前

登录 后才可以发表评论

状态
负责人
项目
Pull Requests
关联的 Pull Requests 被合并后可能会关闭此 issue
预计工期 (小时)
开始日期   -   截止日期
-
置顶选项
优先级
里程碑
分支
参与者(2)
i-robot-I-am-a-robot hujingsong-hu-jingsong
1
https://gitee.com/mindspore/community.git
git@gitee.com:mindspore/community.git
mindspore
community
community

搜索帮助