单张图片预测时间代码
时间: 2023-08-06 07:10:03 浏览: 51
以下是使用PyTorch实现的一个示例代码,用于计算单张图片在预训练的ResNet50模型上的预测时间:
```python
import torch
import torchvision
import time
# 加载预训练的ResNet50模型
model = torchvision.models.resnet50(pretrained=True)
# 加载一张图片
img = torchvision.io.read_image("test.jpg")
# 将图片转换为模型所需的格式
img = torchvision.transforms.functional.resize(img, (224, 224))
img = img.unsqueeze(0) # 添加一个batch维度
# 将模型移至GPU上(如果可用的话)
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model.to(device)
img = img.to(device)
# 对单张图片进行预测,并计算预测时间
start_time = time.time()
with torch.no_grad():
output = model(img)
end_time = time.time()
pred_time = end_time - start_time
print("Prediction time: ", pred_time)
```
在运行上述代码之前,需要先安装PyTorch和torchvision库,并将一张测试图片命名为"test.jpg"放在代码所在的目录下。
相关推荐
![txt](https://img-home.csdnimg.cn/images/20210720083642.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)