今天在翻b站时看到一个up主自己写的pytorch版本的yolo3目标检测框架,觉得很有趣就学习了一下,在此记录下学习中值得注意的点。
大神的讲解视频:https://www.bilibili.com/video/BV1Hp4y1y788?p=1
博客地址:https://blog.youkuaiyun.com/weixin_44791964/article/details/105310627
源代码地址:https://github.com/bubbliiiing/yolo3-pytorch
1.整体代码框架
整体的代码框架如下图,使用的是pytorch框架,易读性很高,使用难度也不大。

2.训练
进行训练时首先要有数据集,这里以VOC2007数据集为例,它是有20个分类的数据集。
当然也可以运用自己制作的数据集,制作自己的数据集可以使用labelimg插件,教程地址:https://www.bilibili.com/video/BV1Hp4y1y788?p=12
step1: 训练前将标签文件放在VO

本文介绍了使用PyTorch实现YOLOv3目标检测框架的详细步骤,包括整体代码结构、训练过程、测试方法及运行结果。通过VOC2007数据集进行训练,利用labelimg工具制作数据集,并提供了训练和测试的具体操作指南。
最低0.47元/天 解锁文章
760

被折叠的 条评论
为什么被折叠?



