TensorRT测试
This commit is contained in:
52
test_480_resolution.py
Normal file
52
test_480_resolution.py
Normal file
@@ -0,0 +1,52 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
480分辨率多摄像头测试脚本
|
||||
测试配置:
|
||||
- 分辨率: 480x480
|
||||
- 批次大小: 8
|
||||
- 测试时长: 120秒
|
||||
- 所有30个摄像头
|
||||
"""
|
||||
|
||||
import subprocess
|
||||
import sys
|
||||
|
||||
def main():
|
||||
print("="*60)
|
||||
print("480分辨率多摄像头性能测试")
|
||||
print("="*60)
|
||||
print("配置:")
|
||||
print(" - 分辨率: 480x480")
|
||||
print(" - 批次大小: 8")
|
||||
print(" - 测试时长: 120秒")
|
||||
print(" - 摄像头数量: 30")
|
||||
print("="*60)
|
||||
print()
|
||||
|
||||
# 运行测试
|
||||
cmd = [
|
||||
sys.executable,
|
||||
"optimized_multi_camera_tensorrt.py",
|
||||
"--config", "config.yaml",
|
||||
"--model", "C:/Users/16337/PycharmProjects/Security/yolo11n.engine",
|
||||
"--batch-size", "8",
|
||||
"--target-size", "480",
|
||||
"--duration", "120"
|
||||
]
|
||||
|
||||
print(f"执行命令: {' '.join(cmd)}")
|
||||
print()
|
||||
|
||||
try:
|
||||
subprocess.run(cmd, check=True)
|
||||
except subprocess.CalledProcessError as e:
|
||||
print(f"\n❌ 测试失败: {e}")
|
||||
return 1
|
||||
except KeyboardInterrupt:
|
||||
print("\n⏹️ 测试被用户中断")
|
||||
return 0
|
||||
|
||||
return 0
|
||||
|
||||
if __name__ == "__main__":
|
||||
sys.exit(main())
|
||||
Reference in New Issue
Block a user