本人最近在利用faster_rcnn训练kitti数据集,其中需要将kitti数据集转为voc数据集,但是发现:
kitti图片是png格式
voc2007是jpg格式
其中有7000多张图片需要批量转换,在网上发现一些代码,但跑起来有错误,于是本人稍作修改:
import os
import string
dirName = "D:your path\\" #最后要加双斜杠,不然会报错
li=os.listdir(dirName)
for filename in li:
newname = filename
newname = newname.split(".")
if newname[-1]=="png":
newname[-1]="jpg"
newname = str.join(".",newname) #这里要用str.join
filename = dirName+filename
newname = dirName+newname
os.rename(filename,newname)
print(newname,"updated successfully")
此方法不仅可以用于转换图片后缀,也可以用于其他文件的后缀转换。