PaddlePaddle第四章学习笔记
这一章讲的是YOLO-V3算法,这是一个检测算法,就是说这个算法不仅仅要解决分类问题,还要检测出物体的位置在哪。先总体概括一下,这个算法就是将图片划分出许多的小方格,再生成一系列的锚框,这些锚框会把整个图片覆盖掉,然后通过锚框与真实框的交并比采用非极大值抑制的方法生成预测框,最后再将图片经过卷积化处理之后与它关联起来,再做一些微调就可以了。
首先要处理数据,这里以林业病虫害数据集为例。如果想去下载此数据集,请点击下列链接:识虫数据集
首先解压数据集,代码如下:
!unzip -d /home/aistudio/work /home/aistudio/data/data19638/insects.zip
交并比就是两个图形的重合程度,如图1
重合程度越高,交并比就越高,如图2
计算loU的代码如下:
# 计算IoU,矩形框的坐标形式为xyxy,这个函数会被保存在box_uti