报错原因:numpy不能读取CUDA tensor 需要将它转化为 CPU tensor。

所以如果想把CUDA tensor格式的数据改成numpy时,需要先将其转换成cpu float-tensor随后再转到numpy格式

报错行:

tcls[index, best_n, g_y_center, g_x_center, np.array(target[index, t, 0])] = 1

修改后:

tcls[index, best_n, g_y_center, g_x_center, np.array(target[index, t, 0].cpu())] = 1

 

相关文章:

  • 2021-12-22
  • 2022-12-23
  • 2022-12-23
  • 2021-12-20
  • 2022-02-24
  • 2022-12-23
  • 2021-08-07
  • 2021-04-14
猜你喜欢
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2021-03-28
  • 2022-02-09
  • 2021-09-08
  • 2022-12-23
相关资源
相似解决方案