加入极市专业CV交流群,与 10000+来自港科大、北大、清华、中科院、CMU、腾讯、百度 等名校名企视觉开发者互动交流!

同时提供每月大咖直播分享、真实项目需求对接、干货资讯汇总,行业技术交流。关注 极市平台 公众号 ,回复 加群,立刻申请入群~

来源|AIWalker

今天在刷github时,突然看到了YOLOv5,笔者当时还在怀疑是不是眼花了?确实时YOLOv5,但不是官方的也不是AB大神版,而是U版YOLO改进版。哎,想想真可怜,笔者还在熟悉YOLOv4的时候,YOLOv5竟然出现了,太快了,跟不上节奏啊。不过不妨碍我们去研究了解它。

笔者之前介绍过YOLOv4,见此处,感兴趣的小伙伴可以去看看。

严格来讲,这里提到的YOLOv4可以视为YOLOv4.5,它仍然是各种tricks的组合,但更关键的是它的实现方式由darknet转为了pytorch,同时还支持ONNX以及CoreML。真乃天大的福音啊,还在等什么么?还不快点去围观关注一波!

YOLOv5代码链接:github.com/ultralytics/ 实现者是谁呢?Ultralytics:ultralytics.com,大家也许听过U版yolo,即:U版YOLOv3 github.com/ultralytics/,见下图拥有4.7k-star的U版YOLOv3,所以你就知道了吧,吼吼


YOLOv5-Ultralytics


言归正传,我们先来看看YOLOv5的一些“官方”说明吧。首先给出了已开源的YOLOv5在COCO2017数据集上的指标,见下图。可以看到,它基本上超过了大部分的目标检测器,同时具有超快的速度。


注1:指的是COCO-test-dev2017上的结果,其他AP表示val2017的精度。所有指标均为单模型、单尺度、为进行任何集成与TTA。可以通过如下命令进行重现上述指标,非常赞啊!

python test.py --img-size 736 --conf_thres 0.001

注2:上述延迟是在V100 GPU上对5000张COCO-val2017数据进行测试的结果,包含数据预处理、模型推理、后处理以及NMS。平均NMS的时间为1.6ms/image。上述指标可以通过如下命令复现:

python test.py --img-size 640 --conf_thres 0.1 --batch-size 16

注3:上述所有模型均采用默认配置与超参数训练了300epoch,且未添加自动增广技术。

下面附上训练曲线与测试效果。



Ultralytics Professional Support

附上一些Ultralytics的信息,摘自:github.com/ultralytics/,为避免凸显笔者粗陋的英语水平,就直接上英文原文了。

Ultralytics is a U.S.-based particle physics and AI startup with over 6 years of expertise supporting government, academic and business clients. We offer a wide range of vision AI services, spanning from simple expert advice up to delivery of fully customized, end-to-end production solutions, including:

  • 「Cloud-based AI」 surveillance systems operating on 「hundreds of HD video streams in realtime.」

  • 「Edge AI」 integrated into custom iOS and Android apps for realtime 「30 FPS video inference.」

  • 「Custom data training」, hyperparameter evolution, and model exportation to any destination.

YOLOv5-updates

最后附上YOLOv5项目的一些更新说明,同上,附上英文原本。

This repository represents Ultralytics open-source research into future object detection methods, and incorporates our lessons learned and best practices evolved over training thousands of models on custom client datasets with our previous YOLO repository github.com/ultralytics/. 「All code and models are under active development, and are subject to modification or deletion without notice.」 Use at your own risk.

Updates:

  • 「May 27, 2020」: Public release of repo. yolov3-spp (this repo) is SOTA among all known yolo implementations, yolov5 family will be undergoing architecture research and development over Q2/Q3 2020 to increase performance. Updates may include CSP bottlenecks from yolov4, as well as PANet or BiFPN head features.

  • 「May 24, 2020」: Training yolov5s/x and yolov3-spp. yolov5m/l suffered early overfitting and also code 137 early docker terminations, cause unknown. yolov5l underperforms yolov3-spp due to earlier overfitting, cause unknown.

  • 「April 1, 2020」: Begin development of a 100% pytorch scaleable yolov3/4-based group of future models, in small, medium, large and extra large sizes, collectively known as yolov5. Models will be defined by new user-friendly yaml-based configuration files for ease of construction and modification. Datasets will likewise use yaml configuration files. New training platform will be simpler use, harder to break, and more robust to training a wider variety of custom dataset.

Appendix

虽然笔者不是专职目标检测方向的工作,但对YOLO还真听过非常多,笔者带过的团队也用过YOLOv3方法进行目标检测相关应用;虽然YOLOv5仍处于研发阶段,尚未如YOLOv4得到官方的认可,但YOLOv5的指标仍然非常棒,也值得各位CVer去研究与应用。最后其他UbanYOLO继续开展后续的研究,感兴趣的CVer还不快起去“素质三连”一波?

推荐阅读:

  • YOLO系列综述:从V1到V4

  • YOLOv4 的各种新实现、配置、测试、训练资源汇总

  • YOLOv3最全复现代码合集(含PyTorch/TensorFlow和Keras等)

添加极市小助手微信(ID : cv-mart),备注:研究方向-姓名-学校/公司-城市(如:AI移动应用-小极-北大-深圳),即可申请加入AI移动应用极市技术交流群,更有每月大咖直播分享、真实项目需求对接、求职内推、算法竞赛、干货资讯汇总、行业技术交流,一起来让思想之光照的更远吧~

△长按添加极市小助手

△长按关注极市平台,获取最新CV干货

觉得有用麻烦给个在看啦~  

onnx 测试_YOLOv5来了!Pytorch实现,支持ONNX和CoreML相关推荐

  1. onnx 测试_ONNX 现场演示教程

    ONNX 现场演示教程 译者:冯宝宝 本教程将向您展示如何使用ONNX将已从PyTorch导出的神经模型传输模型转换为Apple CoreML格式.这将允许您在Apple设备上轻松运行深度学习模型,在 ...

  2. 【目标检测】英雄联盟能用YOLOv5实时目标检测了 支持onnx推理

    目录 一.项目介绍 二.项目结构 三.准备数据 1.数据标注 2.数据转换格式 四.执行训练 1.anchors文件 2.标签文件 3.预训练模型 4.训练数据 5.修改配置 6.执行训练 五.执行预 ...

  3. 【地平线开发板 模型转换】将pytorch生成的onnx模型转换成.bin模型

    文章目录 1 获取onnx模型 2 启动docker容器 3 onnx模型检查 3.1 为什么要检查? 3.2 如何操作 4 图像数据预处理 4.1 一些问题的思考 4.2 图片挑选与放置 4.2 使 ...

  4. Paddle2ONNX最新升级:飞桨模型全面支持ONNX协议啦!

    点击左上方蓝字关注我们 16年前的电影<功夫>可以说是家喻户晓,并将武侠片推向了一个新的高潮.在电影中,街头小混混阿星偶然间被打通了任督二脉,因善开悟,再有"如来神掌" ...

  5. onnx 测试_用于ONNX的TensorRT后端

    用于ONNX的TensorRT后端 解析ONNX模型以使用TensorRT执行. 另请参阅TensorRT文档. 有关最近更改的列表,请参见changelog. 支持的TensorRT版本 Maste ...

  6. 【TensorRT】PyTorch模型转换为ONNX及TensorRT模型

    文章目录 1. PyTorch模型转TensorRT模型流程 2. PyTorch模型转ONNX模型 3. ONNX模型转TensorRT模型 3.1 TensorRT安装 3.2 将ONNX模型转换 ...

  7. Pytorch模型转onnx打包部署(完全脱离pytorch)

    一.问题阐述: 深度学习pytorch框架在实际部署中由于其对环境和资源要求太高以及python语言的限制性,导致其部署存在重重困难,在这里介绍一下onnxruntime(GPU)的部署流程. 二.o ...

  8. TensorRT:在NVIDIA各种GPU硬件平台下运行的高性能推理C++库【Pytorch->ONNX->TensorRT】【实现神经网络提速(读取ONNX模型并运行)】【目前对ONNX的支持最好】

    tensorRT 基本网络构建 创建builder 创建network 创建config network添加输入 network添加层,绑定输入 network mark output,指定输出节点 ...

  9. onnx 测试_Kaldi-ONNX

    Kaldi-ONNX Kaldi-ONNX 是一个将Kaldi的模型文件转换为ONNX模型的工具. 转换得到的ONNX模型可以借助MACE框架部署到Android, iOS, Linux或者Windo ...

最新文章

  1. 网站推广的三大基本方式
  2. python编程入门p-Python是什么?简单了解pythonp-入门
  3. 项目小结之数据库设计
  4. H5新增的标签和属性
  5. hssfworkbook.write 生成文件 数据丢失_电脑文件丢失了,数据恢复我只用这一个软件!...
  6. 寻宝机器人电路板焊接_专业维修淮安市KUKA库卡KRC2机器人回收{机器人调试}
  7. Java NIO和IO的主要区别
  8. html哪个标签显示第一节,HTML第一节
  9. python hook_五分钟内用Python实现GitHook
  10. redis服务器环境下mysql实现lnmp架构缓存
  11. quartus ii IP核的破解
  12. 【测试沉思录】5. 测试人员如何快速熟悉新业务?
  13. python谐音梗_谐音梗生成器
  14. vue前端实现批量导出
  15. PB设置表格背景颜色
  16. Computer:编程入门的简介相关的一些概念解释之详细攻略
  17. 固态硬盘数据恢复教您如何操作
  18. centos7 vim查找_如何在 Vim/Vi 中快速查找和替换文本内容
  19. python 获取网页的内容
  20. word导出html乱码怎么办,Word文档文件变成乱码如何完全修复?-修复百科

热门文章

  1. building for production...Killed
  2. go 排序sort的使用
  3. mybatis强化(一)基本配置补充
  4. frameset标签设计页面
  5. 假期七天实习参观有感
  6. AD域安装及必要设置
  7. [DEncrypt] DESEncrypt--加密/解密帮助类 (转载)
  8. linux之type命令
  9. Hbase+Phoenix使用总结
  10. 开发者须知:关于 Android L 的一切