yolov4-tiny-keras train.py 的81/82行似乎有误

train.py 的81/82行似乎有误

Open cxyth opened this issue 4 years ago • 4 comments

    true_boxes[..., 0:2] = boxes_xy/input_shape[:]
    true_boxes[..., 2:4] = boxes_wh/input_shape[:]

应该是

    true_boxes[..., 0:2] = boxes_xy/input_shape[::-1]
    true_boxes[..., 2:4] = boxes_wh/input_shape[::-1]

input_shape排序是(h, w)，对应上box的xy和wh的话，应该反一下；通常训练正方形数据时无所谓，但在在训练宽高不一致的数据时会溢出

Sep 11 '20 01:09 cxyth

我修改一下

Sep 12 '20 04:09 bubbliiiing

训练非正方形只有这两行有误吗？

Sep 12 '20 04:09 bubbliiiing

训练非正方形只有这两行有误吗？

目前没发现其他问题，训练的效果挺不错

Sep 15 '20 05:09 cxyth

好的

Sep 17 '20 02:09 bubbliiiing