Open WuShaogui opened 4 months ago
https://github.com/lyuwenyu/RT-DETR/blob/2b88d5d53bcbfbb70329bc9c007fdf7e76cf90dc/rtdetr_pytorch/src/zoo/rtdetr/rtdetr_postprocessor.py#L48
改为
scores = F.softmax(logits, dim=-1)
不然最后一个类别无法预测，因为结果被截断了，调试结果如下：
https://github.com/lyuwenyu/RT-DETR/blob/2b88d5d53bcbfbb70329bc9c007fdf7e76cf90dc/rtdetr_pytorch/src/zoo/rtdetr/rtdetr_postprocessor.py#L48
改为
scores = F.softmax(logits, dim=-1)不然最后一个类别无法预测，因为结果被截断了，调试结果如下：