FateScript / CenterNet-better

An easy to understand and better performance version of CenterNet
Apache License 2.0
548 stars 104 forks source link

生成真值index #25

Closed kaiopen closed 4 years ago

kaiopen commented 4 years ago

看起来可以用中文交流? 在 dl_lib/network/generator/centernet_gt.pyCenterNetGT 类 的 generate 方法中,

gt_index[:num_boxes] = centers_int[..., 1] * output_size[0] + centers_int[..., 0]

按理解,output_size[0] 是输出特征图的高度,output_size[1] 是宽度吧?centers_int[1] 是 box 中心点的 y,那这块的 index 计算是不是应该是

centers_int[..., 1] * output_size[1] + centers_int[..., 0]

虽然因为out_size 的高宽一样,对结果没有影响。

FateScript commented 4 years ago

看了一下,的确是有bug的,现在这个地方已经fix了,感谢issue