infinite_jason的博客

训练好一个模型，或找到开源模型后，转为onnx并通过onnxruntime（CPU，GPU）来部署服务。

今天用pytorch处理图像时，涉及到了计算均值的问题，整理一下解决思路。第一种思路：tensor转换为numpy再进行处理。import torchimport cv2img = cv2.imread("image path")tensor_img = torch.from_numpy((img[:, :, ::-1] / 255.0)[None, ...].transpose(0, 3, 1, 2)).cuda()...numpy_img = (tensor_img.de...