2020年5月20日
pytorch转onnx转ncnn问题Unsqueeze not supported yet!
摘要： 1、pytorch模型转onnx input = cv.imread('c:/123.jpg')input = np.transpose(input, (2, 0, 1)).astype(np.float32)now_image1 = Variable(torch.from_numpy(input) 阅读全文
posted @ 2020-05-20 11:45 ~乾坤# 阅读(2734) 评论(0) 推荐(0)
pytorch基于cpu下的速度测试及cpu线程设置
摘要： libtorch默认是将cpu性能全部耗光，cpu利用率达到95%，需要设置并行计算的线程数，设置函数是torch::set_num_threads(1); python版本的torch.set_num_threads(1) resnet50模型 cpu型号：I7 4770 3.6HZ 内存16G 阅读全文
posted @ 2020-05-20 11:32 ~乾坤# 阅读(4200) 评论(0) 推荐(0)