commit 8463f5a57169890be6f378cdd51ac9ffcca0e719 Author: 16337 <1633794139@qq.com> Date: Tue Jan 20 10:49:16 2026 +0800 GPU测试 diff --git a/benchmark/__init__.py b/benchmark/__init__.py new file mode 100644 index 0000000..78bf97f --- /dev/null +++ b/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", +]