Tencent / ObjectDetection-OneStageDet

单阶段通用目标检测器

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

A problem about anchor setting

LDoubleZhi opened this issue · comments

  yolov3用自己的数据集训练时,设置anchor时候需要尺度不变放缩到一个input_size然后聚类得到anchor吗?比如我都resize到640*640,然后kmeans聚类得anchor.
  但实际上我debug时候发现源码训练里加入了random_input_size,那这样的话,上面设置的anchor_size就会出现不匹配的情况了(训练时候输入尺度是随机的).

所以应该用多尺度数据聚类anchor吗?

是否应该先按一个尺度把自己的数据集聚类得到的anchor_size作为参照,然后在random_input_scale时按参照的尺度放缩anchor_size来训练

你搞混了哦,从yoloV2开始,就设置了随机尺度缩放,每训练一定的steps,yaml文件配置的是lr_steps: [400,700,900,1000, 40000,45000],将输入图片尺寸缩放32*(10,19)对应着[320,608]个随机尺寸进行训练,这是为了适应不同的图片尺寸,增加鲁棒性。和anchor没关系,anchor对应的是小中大物体。anchor聚类是作者根据数据集中所有的w和h,设置分类(v2是5 v3是9)个数聚类来的。

请问有kmeans代码吗