yolov5ds icon indicating copy to clipboard operation
yolov5ds copied to clipboard

分割

Open autumn-qt opened this issue 3 years ago • 7 comments

大佬你好,请问你这个里面分割是用的什么算法?

autumn-qt avatar Feb 24 '22 11:02 autumn-qt

直接在yolov5的neck后面接一个分割头用于分割

midasklr avatar Feb 28 '22 07:02 midasklr

这个分割头是用的那种算法的分割头pspnet还是deeplab的

autumn-qt avatar Feb 28 '22 08:02 autumn-qt

类似encode-decode,结构如下:

https://github.com/midasklr/yolov5ds/blob/ba0809f0c05b425bcb348f0aa63af2eca429ca82/models/segheads.yaml#L5-L15

midasklr avatar Mar 01 '22 06:03 midasklr

大佬,能否把你的分割数据集百度云分享一下,不太了解你这数据集里面的结构是什么样的。

autumn-qt avatar Mar 04 '22 13:03 autumn-qt

请问这个分割和检测是一起做的吗,训练为什么要前向传播两次呢?一次得到目标检测结果?一次得到分割结果?

HuKai97 avatar Mar 28 '22 03:03 HuKai97

训练的时候为啥要分两次训呢?可以一起训不?

HuKai97 avatar Apr 01 '22 12:04 HuKai97

请问这个分割和检测是一起做的吗,训练为什么要前向传播两次呢?一次得到目标检测结果?一次得到分割结果?

在readme中有:因为VOC数据集并没有同时提供所有的检测和分割标注,改为一个batch的检测和一个batch的分割数据集迭代训练,也即是训练时,一个batch检测和分割数据进行bp,然后梯度累积,然后统一进行更新参数.

ganleiboy avatar Jun 08 '22 09:06 ganleiboy