import os
os.chdir('XXX') #数据集存储路径
for i, img in enumerate(os.listdir()):
name = str(i)
ext = os.path.splitext(img)[-1] #获取文件类型后缀
os.rename(img,f"{name}{ext}")
效果
import os
os.chdir('XXX') #数据集存储路径
for i, img in enumerate(os.listdir()):
name = str(i)
ext = os.path.splitext(img)[-1] #获取文件类型后缀
os.rename(img,f"{name}{ext}")
效果