50 lines
1.0 KiB
Python
50 lines
1.0 KiB
Python
__all__ = ['extensions']
|
|
|
|
class model_extensions:
|
|
def __init__(self, model, model_data, model_quantize, input_meta, output_meta):
|
|
self._model = model
|
|
self._model_data = model_data
|
|
self._model_quantize = model_quantize
|
|
self._input_meta = input_meta
|
|
self._output_meta = output_meta
|
|
|
|
@property
|
|
def model(self):
|
|
return self._model
|
|
|
|
@property
|
|
def model_data(self):
|
|
return self._model_data
|
|
|
|
@property
|
|
def model_quantize(self):
|
|
return self._model_quantize
|
|
|
|
@property
|
|
def input_meta(self):
|
|
return self._input_meta
|
|
|
|
@property
|
|
def output_meta(self):
|
|
return self._output_meta
|
|
|
|
class file_model:
|
|
def __init__(self,extensions):
|
|
self._extensions = extensions
|
|
|
|
@property
|
|
def extensions(self):
|
|
return self._extensions
|
|
|
|
x_extensions = model_extensions(
|
|
'.json',
|
|
'.data',
|
|
'.quantize',
|
|
'_inputmeta.yml',
|
|
'.yml'
|
|
)
|
|
|
|
_file_model = file_model(x_extensions)
|
|
|
|
extensions = _file_model.extensions
|