Python + OpenCV 深度学习 数据集 批量规范为 .jpg

#
# Created by foreverstrong
#

import cv2
import os
import numpy as np
import sys

print("\nsys.version:")
print(sys.version)

image_path_nojpg = "/home/strong/demo_workspace/rename_file/image_data"


image_dir_nojpg = image_path_nojpg + '/'
filepath = os.listdir(image_dir_nojpg)

# filepath.sort()

image_path_jpg = image_path_nojpg + "_jpg"
image_dir_jpg = image_path_jpg + '/'

if not os.path.exists(image_dir_jpg):
    os.makedirs(image_dir_jpg)

num = 0

for imagename in filepath:

    image_path_name = image_dir_nojpg + imagename

    img = cv2.imread(image_path_name.strip())
    # print(image_path_name.strip())
    imgcopy = img.copy()

    img_height, img_width, img_channel = img.shape

    padnum = str(num + 1).zfill(6)
    newname = "" + padnum + '.jpg'

    cv2.imwrite("%s%s" % (image_dir_jpg, newname), imgcopy)
    num = num + 1

    cv2.imshow("image_show", imgcopy)
    keyboard = cv2.waitKey(10) & 0xFF

    # wait for ESC key to exit
    if keyboard == 27:
        break

cv2.destroyAllWindows()

/usr/bin/python2.7 /home/strong/demo_workspace/rename_file/nojpg_to_jpg.py

sys.version:
2.7.12 (default, Dec  4 2017, 14:50:18) 
[GCC 5.4.0 20160609]

Process finished with exit code 0