RuntimeError: selected index k out of range

[Kingstom8MB]

我为了提高速度，检测图片前将目标区域裁剪下来，但目标区域似乎太小了，出现了这个错误。请问有修复方法吗？如果我直接全局识别图片的话就没有这个问题

[Kingstom8MB]

错误发生在第49行：scores, indices = ((hm == hm_pool).float() * hm).view(1, -1).cpu().topk(1000)，detect函数里

[RiceZ]

我也遇到了同样的问题，图像太小会报这个错误。
报错时可以看下hm和hm_pool尺寸大小，应该是小于1000的，所以topk(1000)就超出索引了。
把topk中的参数改小就好了：
scores, indices = ((hm == hm_pool).float() * hm).view(1, -1).cpu().topk(100)