GPU测试

benchmark/__init__.py

@@ -0,0 +1,40 @@
+"""
+FP16 性能评估 Benchmark 框架
+
+用于评估 RTX 3050 GPU 在 YOLOv11n 模型下的 FP16 推理性能极限。
+"""
+
+__version__ = "1.0.0"
+__author__ = "Edge AI System Engineer"
+
+from .config import BenchmarkConfig, load_config, save_config
+from .engine_builder import TRTEngineBuilder, EngineMetadata
+from .inference_engine import TRTInferenceEngine
+from .decode_thread import DecodeThread, FrameQueueManager
+from .batch_assembler import BatchAssembler, GPUPreprocessor
+from .metrics_collector import MetricsCollector, GPUMonitor
+from .results import TestResult, export_json, export_csv, generate_report
+from .benchmark_runner import BenchmarkRunner
+from .stress_test import StressTestRunner, run_stress_test
+
+__all__ = [
+    "BenchmarkConfig",
+    "load_config",
+    "save_config",
+    "TRTEngineBuilder",
+    "EngineMetadata",
+    "TRTInferenceEngine",
+    "DecodeThread",
+    "FrameQueueManager",
+    "BatchAssembler",
+    "GPUPreprocessor",
+    "MetricsCollector",
+    "GPUMonitor",
+    "TestResult",
+    "export_json",
+    "export_csv",
+    "generate_report",
+    "BenchmarkRunner",
+    "StressTestRunner",
+    "run_stress_test",
+]