1 import os

 2 import cv2

 3 class GeneDataset():

 4     def __init__(self,rootdir):

 5         self.rootdir = rootdir

 6         self.subdirs = []

 7     def looksubdir(self):

 8         list_dirs = os.walk(self.rootdir)

 9         for root, dirs, files in list_dirs:

10             for d in dirs:

11                 self.subdirs.append(os.path.join(root, d))

12                 print("subdir =", os.path.join(root, d))

13     def reformat(self):14         for subdir in self.subdirs:

15             list_dirs = os.walk(subdir)

16             for root, dirs, files in list_dirs:

17                 for f in files:

18                     srcname = os.path.join(root, f)

19                     print('srcname', srcname)

20                     srcformat = srcname.split('.')[-1]

21                     if srcformat is not '.jpg':

22                         img = cv2.imread(srcname)

23                         newname = srcname.replace(srcformat, 'jpg')

24                         print('newname=', newname)

25                         cv2.imwrite(newname, img)

26                         os.remove(srcname)

27 path = r'E:\data\pic'

28 myclassdataset = GeneDataset(path)

29 myclassdataset.looksubdir()

30 myclassdataset.reformat()