def init_engine_sdk(self,
engine_type,
local_engine_port="53001",
timeout=60):
local_engine_port = os.environ.get("LOCAL_ENGINE_PORT",
local_engine_port)
if local_engine_port is None:
local_engine_port = "53001"
if isInCloudMode():
response = platform_helper.get_platform_client().platform_forward(
self.__device_sdk_port)
if response is None:
return ERR_CLOUD_PLATFORM
else:
local_engine_port = response["localPort"]
else:
self.__forward(local_engine_port, self.__device_sdk_port)
counts = int(timeout / 2 + 1)
for i in range(counts):
try:
self.__engine_connector = EngineFactory.createEngine(
engine_type, os.environ.get("PLATFORM_IP", "127.0.0.1"),
int(local_engine_port))
if self.__engine_connector is None:
logger.error("create engine failed . invalid type : " +
str(engine_type))
return ERR_INVALID_ENGINETYPE
version = self.__engine_connector.get_sdk_version()
if version:
logger.debug(version)
return ERR_SUCCEED
except Exception as e:
time.sleep(2)
logger.error("init engine sdk timeout")
return ERR_CONNECT_TO_SDK_FAILED
def __init__(self):
self.errs = []
self.image_id = 10000
self.platform_client = platform_helper.get_platform_client()