46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
"""
|
|
RTX 3050 GPU 压力测试启动脚本
|
|
|
|
测试内容：
|
|
1. 不同分辨率下最大每秒处理帧数
|
|
2. 最大接入摄像头路数
|
|
3. 不同摄像头数量下单路最大帧数
|
|
4. 不同抽帧率下最大摄像头数
|
|
|
|
运行方式：
|
|
python run_stress_test.py
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
|
|
# 添加当前目录到路径
|
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
from benchmark.stress_test import run_stress_test
|
|
|
|
if __name__ == "__main__":
|
|
# 模型路径
|
|
MODEL_PATH = "C:/Users/16337/PycharmProjects/Security_project/yolov8n.pt"
|
|
|
|
# 输出目录
|
|
OUTPUT_DIR = "./stress_results"
|
|
|
|
print("=" * 60)
|
|
print("RTX 3050 GPU 压力测试")
|
|
print("=" * 60)
|
|
print(f"模型: {MODEL_PATH}")
|
|
print(f"输出: {OUTPUT_DIR}")
|
|
print()
|
|
print("测试内容:")
|
|
print(" 1. 最大处理帧数 (320x320, 480x480)")
|
|
print(" 2. 最大摄像头接入数")
|
|
print(" 3. 不同摄像头数量下单路最大帧数")
|
|
print(" 4. 不同抽帧率下最大摄像头数")
|
|
print()
|
|
print("预计时间: 20-30 分钟")
|
|
print("按 Ctrl+C 可随时中断，已完成的结果会自动保存")
|
|
print("=" * 60)
|
|
|
|
run_stress_test(MODEL_PATH, OUTPUT_DIR)
|