Skip to content

GitLab

Commit 33ca217f authored by Niyaz's avatar Niyaz Committed by Neal Wu
Browse files

Fixing the filename check for PNG files (#2450)

parent 16e46798
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
research/inception/inception/data/build_image_data.py
View file @ 33ca217f
...@@ -183,7 +183,7 @@ def _is_png(filename): ...@@ -183,7 +183,7 @@ def _is_png(filename):
Returns: Returns:
boolean indicating if the image is a PNG. boolean indicating if the image is a PNG.
""" """
return '.png' in filename return filename.endswith('.png')
def _process_image(filename, coder): def _process_image(filename, coder):
...@@ -264,7 +264,7 @@ def _process_image_files_batch(coder, thread_index, ranges, name, filenames, ...@@ -264,7 +264,7 @@ def _process_image_files_batch(coder, thread_index, ranges, name, filenames,
image_buffer, height, width = _process_image(filename, coder) image_buffer, height, width = _process_image(filename, coder)
except Exception as e: except Exception as e:
print(e) print(e)
print('SKIPPED: Unexpected eror while decoding %s.' % filename) print('SKIPPED: Unexpected error while decoding %s.' % filename)
continue continue
example = _convert_to_example(filename, image_buffer, label, example = _convert_to_example(filename, image_buffer, label,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment