One-YOLOv5 v1.2.0发布:支持分类、检测、实例分割

One-YOLOv5 v1.2.0同步Ultralytics YOLOv5 v7.0,支持分类、检测和实例分割任务。新特性包括flask_rest_api支持、wandb实验跟踪、性能优化等。针对COCO数据集,展示了 yolov5s-default 的精度和单GPU性能。常见问题如满屏误检框、ONNX模型导出等也得到了解答。未来计划优化训练速度和显存占用,增加CPU ONNX模型导出及autobatch功能。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

0c15167e7b8765bcd720b3846731930c.jpeg
 

One-YOLOv5 v1.2.0正式发布。完整更新列表请查看链接:https://github.com/Oneflow-Inc/one-yolov5/releases/tag/v1.2.0,欢迎体验新版本,期待你的反馈。

1

新版本特性

1. 同步了Ultralytics YOLOv5的上游分支v7.0,同时支持分类、目标检测、实例分割任务

2. 支持flask_rest_api

3. 支持使用 wandb 对实验跟踪和可视化功能 

4. oneflow_hub_support_pilimage 

5.为每个batch的compute_loss部分减少一次h2d和cpu slice_update操作 

6. 优化 bbox_iou 函数和模型滑动平均部分,大幅提升训练性能 (一块RTX 3090加速训练YOLOv5s,时间减少11个小时,速度提升20%)

7. 兼容FlowFlops,训练时可以展示模型的FLOPs 

(基于 Flowflops 详解深度学习网络的 FLOPs 和 MACs 计算方案)

84d178ee3c660d7b9dc6e59c5538fe41.jpeg 1249262d056ca043dc886b21b5713eb9.jpeg
b8235a2eaecbdeea7eb56891f905b215.jpeg 142bd092b500aaf47f04cff662b4d31a.jpeg

原图

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值