One-YOLOv5 v1.2.0 发布:支持分类、检测、实例分割
One-YOLOv5 v1.2.0 正式发布。完整更新列表请查看链接:https://github.com/Oneflow-Inc/one-yolov5/releases/tag/v1.2.0,欢迎体验新版本,期待你的反馈。 1 新版本特性 1.同步了 UltralyticsYOLOv5的上游分支 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 计算方案) ...


