File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -48,10 +48,6 @@ def scan_subdir(model_type):
4848 if filename .endswith ('.pt' ):
4949 available_models .append (filename )
5050 data_config_avai [filename ] = scan_data_config (filename [:- 3 ])
51- elif model_type == 'tensorflow' :
52- if filename .endswith ('.pb' ):
53- available_models .append (filename )
54- data_config_avai [filename ] = scan_data_config (filename [:- 3 ])
5551 elif model_type == 'onnx' :
5652 if filename .endswith ('.onnx' ):
5753 available_models .append (filename )
@@ -62,7 +58,7 @@ def scan_subdir(model_type):
6258def scan ():
6359 """Scan all directories that contain example models and write to a file."""
6460
65- available_types = ['keras' , 'pytorch' , 'onnx' , 'tensorflow' ]
61+ available_types = ['keras' , 'pytorch' , 'onnx' ]
6662
6763 #Dictionary to save available model list
6864 model_dict = {}
@@ -80,4 +76,4 @@ def scan():
8076 json .dump (model_dict , open ("available_models.json" , 'w' ))
8177 json .dump (data_config_avai , open ("available_data_config.json" , 'w' ))
8278
83- scan ()
79+ scan ()
You can’t perform that action at this time.
0 commit comments