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