"""
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)