tiny-yolo 训练自己的数据集

感谢团队里面小伙伴们。

tiny-yolo 使用自己的数据集进行训练

./darknet partial cfg/darknet.cfg darknet.weights darknet.conv.10 10

./darknet detector train cfg/voc.data cfg/tiny-yolo-voc.cfg darknet.weights

下载

1.darknet.cfg https://pjreddie.com/darknet/imagenet/#extraction
这里写图片描述
点击cfg file
选择原因:较小的模型,适用于tiny-yolo的速度快

运行

./darknet partial cfg/darknet.cfg darknet.weights darknet.conv.10 10

选择10的原因:tiny-yolo 卷积层数为9,yolo 卷积层数为22(参考得出)

训练

./darknet detector train cfg/voc.data cfg/tiny-yolo-voc.cfg darknet.weights

采用刚刚生成的预训练模型进行训练

### Tiny-YOLO 模型概述 Tiny-YOLOYOLO (You Only Look Once) 家族中的轻量化版本,旨在提供更快的速度以适应资源受限环境下的实时应用需求。相较于标准版 YOLOTiny-YOLO 减少了网络层数和滤波器数量来降低计算复杂度,从而提高了推理速度并减少了内存占用[^4]。 对于希望快速部署或在嵌入式设备上运行的目标检测任务来说,Tiny-YOLO 提供了一个很好的折衷方案,在保持一定精度的同时显著提升了处理效率。例如,在边缘计算场景下,CSL-YOLO 这样的改进模型能够在减少约 57% 的浮点运算次数(FLOPs) 和 参数量的情况下超越 TinyYOLOv4 的表现[^3]。 ### 使用方法 为了方便开发者使用 Tiny-YOLO 模型,社区提供了多种框架支持以及详细的教程文档: - **PyTorch 实现**: 可通过访问特定 GitHub 仓库获取完整的从训练到部署指南,包括但不限于如何准备数据集、调整超参数设置等内容[^2]。 - **PaddlePaddle 集成**: 对于偏好 PaddlePaddle 平台的用户而言,官方给出了基于此平台实现垃圾图像识别的具体步骤说明,涵盖了从数据收集整理直至最终上线预测服务全流程指导。 具体操作过程中需要注意的是,尽管 Tiny-YOLO 设计之初就考虑到了易用性和高效性之间的平衡,但在实际应用场景中仍需根据具体的硬件条件和个人需求做出适当优化调整。 ```python import torch from models import tiny_yolo model = tiny_yolo.TinyYoloNet() input_tensor = torch.randn(1, 3, 416, 416) output = model(input_tensor) print(output.shape) ```
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值