Stinky-Tofu / Stronger-yolo

🔥Improve yolo with latest paper
MIT License
3 stars 0 forks source link

关于 数据读取 的问题 #100

Open ghoshaw opened 4 years ago

ghoshaw commented 4 years ago

Hi, 在数据读取中,也就是__create_label 函数中,写到“ 只要某个GT落入grid中,那么这个grid就负责预测它,最多负责预测gt_per_grid个GT,那么该grid中对应位置的数据为(xmin, ymin, xmax, ymax, 1, classes, mixup_weights), 其他grid对应位置的数据都为(0, 0, 0, 0, 0, 0..., 1)”, 但是我看代码,当gt_count=0的时候,会把当前位置(yind, xind)所有的__gt_per_grid 个label都先置零,然后全部赋值 当前的bbox_label。 如果没有另一个gt落在这个位置了,那么 当前位置 的 gt_per_grid 个 label 都是一样的,而不是有一个gt的label,其他为 00001,我的理解对吗?

Stinky-Tofu commented 4 years ago

哈哈,以代码为准,每次改了代码之后,都忘记改注释了。 @ghoshaw

Stinky-Tofu commented 4 years ago

有什么问题可以加我微信 18890099926 @ghoshaw