41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
"""
|
|
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",
|
|
]
|