CenterNet-Lite icon indicating copy to clipboard operation
CenterNet-Lite copied to clipboard

Published 20 hours ago • verified publisher icon yjh0410

您好，请问边框解码的这段代码怎么解读呀

Open ganyibo opened this issue 4 years ago • 1 comments

topk_scores = topk_scores[0].cpu().numpy() topk_ind = topk_clses[0].cpu().numpy() topk_bbox_pred = bbox_pred[topk_inds[0]].cpu().numpy() 为什么要去取[0]啊，我不太理解

ganyibo avatar Nov 25 '20 07:11 ganyibo

这个[0]是batch的维度，测视的时候，就输入进来一张图片嘛，不需要batch这个维度，就干脆在这把它取掉了，没有其他的含义

yjh0410 avatar Dec 14 '20 05:12 yjh0410