35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
import inspect
|
|
import os
|
|
|
|
def get_modelfile_name(path):
|
|
suffix_list = ['.prototxt', '.pb', '.tflite', '.cfg', '.onnx', '.pt', '.h5', '.json', '.data']
|
|
for file in os.listdir(path):
|
|
if os.path.isfile(os.path.join(os.path.abspath(path), file)):
|
|
modelfile_name, suffix = os.path.splitext(file)
|
|
tmp = modelfile_name.split('.')[-1]
|
|
if suffix in suffix_list and tmp != 'quantize':
|
|
return modelfile_name
|
|
|
|
def print_params(func, **kwargs):
|
|
params = {}
|
|
sig = inspect.signature(func)
|
|
for name, param in sig.parameters.items():
|
|
if name == 'kwargs' or name == 'net':
|
|
continue
|
|
params[name] = param.default
|
|
if kwargs is not None:
|
|
for k, d in kwargs.items():
|
|
params[k] = d
|
|
print_dict(params)
|
|
|
|
def print_dict(dict):
|
|
print_str = "Prameters: ("
|
|
for key, val in dict.items():
|
|
print_str = print_str + str(key) + "="
|
|
if isinstance(val, str):
|
|
print_str = print_str + "\'" + str(val) + "\'"
|
|
else:
|
|
print_str = print_str + str(val)
|
|
print_str = print_str + ', '
|
|
print(print_str.strip(", ") + ")")
|