cv2读取图片,放到gpu中案例
时间: 2023-04-03 19:04:21 浏览: 368
可以使用以下代码将图片读取并放到GPU中:
```python
import cv2
import numpy as np
import torch
# 读取图片
img = cv2.imread('image.jpg')
# 将图片转换为numpy数组
img = np.array(img)
# 将numpy数组转换为torch张量
img_tensor = torch.from_numpy(img)
# 将张量放到GPU中
img_tensor = img_tensor.cuda()
```
这样,图片就被成功地读取并放到了GPU中。
阅读全文