GPU测试

run_stress_test.py

@@ -0,0 +1,45 @@
+"""
+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)