Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
cxyth opened this issue Sep 11, 2020 · 4 comments
Open

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

cxyth opened this issue Sep 11, 2020 · 4 comments

Comments

@cxyth
Copy link

cxyth commented Sep 11, 2020

    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的话,应该反一下;通常训练正方形数据时无所谓,但在在训练宽高不一致的数据时会溢出

@bubbliiiing
Copy link
Owner

我修改一下

@bubbliiiing
Copy link
Owner

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

@cxyth
Copy link
Author

cxyth commented Sep 15, 2020

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

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

@bubbliiiing
Copy link
Owner

好的

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants