今天在翻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: 训练前