视频分类 r2plus1d 推理测试

r2plus1d 推理测试

输入是224*224，推理速度25ms

import torch
import time
from pytorchvideo.models import r2plus1ddef val_inference_speed_simple(device="cuda", iters=30):"""最简单的推理速度测试函数"""device = torch.device(device if torch.cuda.is_available() else "cpu")model = r2plus1d.create_r2plus1d(dropout_rate=0.5).to(device).eval()input_tensor = torch.randn(1, 3, 16, 224, 224).to(device)# 预热with torch.no_grad():for _ in range(10):_ = model(input_tensor)# 测试start_time = time.time()with torch.no_grad():for _ in range(iters):_ = model(input_tensor)if device.type == 'cuda':torch.cuda.synchronize()total_time = time.time() - start_timeavg_latency = (total_time / iters) * 1000fps = 1000 / avg_latencyprint(f"平均延迟: {avg_latency:.2f} ms")print(f"推理速度: {fps:.2f} FPS")return avg_latency, fps# 使用
avg_latency, fps = val_inference_speed_simple()