报错原因: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.…