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

gt label的问题 #10

Open
elvindp opened this issue Oct 29, 2021 · 1 comment
Open

gt label的问题 #10

elvindp opened this issue Oct 29, 2021 · 1 comment

Comments

@elvindp
Copy link

elvindp commented Oct 29, 2021

在'solov2_head.py'里,'solov2_target_single'函数会生成ground truth label,
cate_label初始值是0:
cate_label = torch.zeros([num_grid, num_grid], dtype=torch.int64, device=device)
然后在下边根据中心点的位置,更新cate_label:
cate_label[top:(down+1), left:(right+1)] = gt_label
但class是从0-80,第0类是人,物体很多时候并不是布满整幅图的,这样会cate label里就有大量的0类
有人可以帮忙解释一下不?谢谢

@ViatorSun
Copy link

第0类应该是背景吧,在config.py 文件里面有映射,person 对应标签1

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