z1069614715 / objectdetection_script

一些关于目标检测的脚本的改进思路代码,详细请看readme.md
5.3k stars 476 forks source link

OTA报错 #6

Closed WangRongsheng closed 1 year ago

WangRongsheng commented 1 year ago
  File "/home/gem/utils/loss_OTA.py", line 291, in __call__
    tobj[b, a, gj, gi] = (1.0 - self.gr) + self.gr * iou.detach().clamp(0).type(tobj.dtype)  # iou ratio
RuntimeError: shape mismatch: value tensor of shape [57, 1] cannot be broadcast to indexing result of shape [57]
z1069614715 commented 1 year ago

pytorch版本要求大于12.1+

WangRongsheng commented 1 year ago

已解决,请使用cuda