asnumpy()的速度太慢了 求优化

我在cpu上跑 至少要1s。。。这个太耗时间了 有没有打算优化一下的

你是这么测的时间么, 如果没写nd.waitall(), 一般会把延迟初始化和forward时间计算进来.

output = net(data)
nd.waitall()
t = time.time()
output = output.asnumpy()
print(time.time() - t)
1赞

哦哦哦好的 我刚从其他框架转来 我去试试

class_IDs, scores, bounding_boxs = net(x)
nd.waitall()
img = draw_bbox(img, bounding_boxs[0], scores[0], class_IDs[0])

似乎没起作用 draw_bbox中调用了asnumpy()

可以了 感谢!