administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-03 17:55:21 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Add real amd GPU stats

Browse Source
This commit is contained in:
Nick Mowen 2022-09-24 17:59:22 -06:00
parent 497ebf63f2
commit 38317b9ed4
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
frigate/stats.py
View File

@ -13,6 +13,7 @@ from frigate.comms.dispatcher import Dispatcher
from frigate.config import FrigateConfig from frigate.config import FrigateConfig
from frigate.const import DRIVER_AMD, DRIVER_ENV_VAR, RECORD_DIR, CLIPS_DIR, CACHE_DIR from frigate.const import DRIVER_AMD, DRIVER_ENV_VAR, RECORD_DIR, CLIPS_DIR, CACHE_DIR
from frigate.types import StatsTrackingTypes, CameraMetricsTypes from frigate.types import StatsTrackingTypes, CameraMetricsTypes
from frigate.util import get_amd_gpu_stats
from frigate.version import VERSION from frigate.version import VERSION
from frigate.util import get_cpu_stats from frigate.util import get_cpu_stats
from frigate.object_detection import ObjectDetectProcess from frigate.object_detection import ObjectDetectProcess
@ -98,15 +99,16 @@ def get_gpu_stats(config: FrigateConfig) -> dict[str, str]:
elif "qsv" in args: elif "qsv" in args:
# intel QSV GPU # intel QSV GPU
gpu["name"] = "intel-qsv" gpu["name"] = "intel-qsv"
gpu["usage"] = "100" gpu["gpu_usage"] = "100"
gpu["memory"] = "200" gpu["memory_usage"] = "200"
elif "vaapi" in args: elif "vaapi" in args:
driver = os.environ.get(DRIVER_ENV_VAR) driver = os.environ.get(DRIVER_ENV_VAR)
if driver == DRIVER_AMD: if driver == DRIVER_AMD:
gpu["name"] = "amd-vaapi" amd_usage = get_amd_gpu_stats()
gpu["usage"] = "100"
gpu["memory"] = "200" if amd_usage:
stats["amd-vaapi"] = amd_usage
else: else:
gpu["name"] = "intel-vaapi" gpu["name"] = "intel-vaapi"
gpu["usage"] = "100" gpu["usage"] = "100"
@ -114,8 +116,6 @@ def get_gpu_stats(config: FrigateConfig) -> dict[str, str]:
elif "v4l2m2m" in args: elif "v4l2m2m" in args:
gpu["name"] = "RPi" gpu["name"] = "RPi"
stats["name"] = "name"
return stats return stats