IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix input files parsing (#8718)

Browse Source
This commit is contained in:
Alexey Lebedev 2021-11-22 12:15:00 +03:00 committed by GitHub
parent a33b75871d
commit 8e46eecac0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tools/benchmark_tool/openvino/tools/benchmark/utils/inputs_filling.py
View File

@ -134,10 +134,10 @@ def get_files_by_extensions(paths_to_input, extensions):
check_files_exist(files)
return files
return get_files_by_extensions_for_not_list_of_files(paths_to_input, extensions)
return get_files_by_extensions_for_directory_or_list_of_files(paths_to_input, extensions)
def get_files_by_extensions_for_not_list_of_files(paths_to_input, extensions):
def get_files_by_extensions_for_directory_or_list_of_files(paths_to_input, extensions):
input_files = list()
for path_to_input in paths_to_input:
@ -150,8 +150,8 @@ def get_files_by_extensions_for_not_list_of_files(paths_to_input, extensions):
file_extension = get_extension(file)
if file_extension in extensions:
input_files.append(file)
input_files.sort()
return input_files
input_files.sort()
return input_files
def get_extension(file_path):