cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update base_options.py

Browse Source
This commit is contained in:
Kinar R 2023-05-18 18:09:39 +05:30 committed by GitHub
parent f63baaf8d2
commit 6100f0e76e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
mediapipe/tasks/python/core/base_options.py
View File

@ -16,6 +16,7 @@
import dataclasses import dataclasses
import enum import enum
import os import os
import platform
from typing import Any, Optional from typing import Any, Optional
from mediapipe.calculators.tensor import inference_calculator_pb2 from mediapipe.calculators.tensor import inference_calculator_pb2
@ -63,10 +64,22 @@ class BaseOptions:
else: else:
full_path = None full_path = None
if self.delegate == BaseOptions.Delegate.GPU: platform = platform.system()
acceleration_proto = _AccelerationProto(gpu=_DelegateProto.Gpu())
if self.delegate is not None:
if platform == "Linux":
if self.delegate == BaseOptions.Delegate.GPU:
acceleration_proto = _AccelerationProto(gpu=_DelegateProto.Gpu())
else:
acceleration_proto = _AccelerationProto(tflite=_DelegateProto.TfLite())
elif platform == "Windows":
raise Exception("Delegate is unsupported for Windows.")
elif platform == "Darwin":
raise Exception("Delegate is unsupported for MacOS.")
else:
raise Exception("Unidentified system")
else: else:
acceleration_proto = _AccelerationProto(tflite=_DelegateProto.TfLite()) acceleration_proto = None
return _BaseOptionsProto( return _BaseOptionsProto(
model_asset=_ExternalFileProto( model_asset=_ExternalFileProto(