Fix classification and move segmentation

This commit is contained in:
jiangjiajun
2022-08-11 13:21:42 +00:00
parent 5c6b3d6150
commit d557be18ba
12 changed files with 126 additions and 53 deletions

View File

@@ -19,7 +19,8 @@ from . import c_lib_wrap as C
class Runtime:
def __init__(self, runtime_option):
self._runtime = C.Runtime()
assert self._runtime.init(runtime_option), "Initialize Runtime Failed!"
assert self._runtime.init(
runtime_option._option), "Initialize Runtime Failed!"
def infer(self, data):
assert isinstance(data, dict), "The input data should be type of dict."