fix error when preprocessing image data

Author: pythonlessons

Tutorials/02_captcha_to_text/train.py

@@ -34,7 +34,7 @@ def download_and_unzip(url, extract_to='Datasets'):
 captcha_path = os.path.join('Datasets', 'captcha_images_v2')
 for file in os.listdir(captcha_path):
     file_path = os.path.join(captcha_path, file)
-    file_name = os.path.basename(file)
+    file_name = os.path.splitext(file)[0]
     dataset.append([file_path, file_name])
     vocab.update(list(file_name))
     max_len = max(max_len, len(file_name))