|
@@ -8,11 +8,18 @@ from PIL.Image import Image as PILImage
|
|
|
|
|
|
|
|
|
class BaseSession:
|
|
|
- def __init__(self, model_name: str, sess_opts: ort.SessionOptions, providers=None, *args, **kwargs):
|
|
|
+ def __init__(
|
|
|
+ self,
|
|
|
+ model_name: str,
|
|
|
+ sess_opts: ort.SessionOptions,
|
|
|
+ providers=None,
|
|
|
+ *args,
|
|
|
+ **kwargs
|
|
|
+ ):
|
|
|
self.model_name = model_name
|
|
|
-
|
|
|
+
|
|
|
self.providers = []
|
|
|
-
|
|
|
+
|
|
|
_providers = ort.get_available_providers()
|
|
|
if providers:
|
|
|
for provider in providers:
|
|
@@ -20,7 +27,7 @@ class BaseSession:
|
|
|
self.providers.append(provider)
|
|
|
else:
|
|
|
self.providers.extend(_providers)
|
|
|
-
|
|
|
+
|
|
|
self.inner_session = ort.InferenceSession(
|
|
|
str(self.__class__.download_models()),
|
|
|
providers=self.providers,
|