Commit 5db19735 authored by derekjchow's avatar derekjchow Committed by GitHub
Browse files

Merge pull request #2340 from lime62/fix_code_doc

Fix code documentation.
parents b7ce045c 279305ee
......@@ -55,7 +55,7 @@ def get_class_name_from_filename(file_name):
ie. "american_pit_bull_terrier_105.jpg"
Returns:
example: The converted tf.Example.
A string of the class name.
"""
match = re.match(r'([A-Za-z_]+)(_[0-9]+\.jpg)', file_name, re.I)
return match.groups()[0]
......
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