OpenCV—python图像标注（批量文件重命名）

原创已于 2023-04-05 09:36:36 修改 · 3k 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#OpenCV—python图像标注（批量文件重命名）

于 2018-09-26 11:33:46 首次发布

OpenCV/python 专栏收录该内容

84 篇文章

订阅专栏

本文介绍了一种使用Python批量修改图片文件名的方法，适用于图片数据集的预处理阶段，如为图片添加统一前缀并按序编号，有助于提高数据管理效率。


"""
批量修改文件名：我这里用于对图片数据集标注
"""
import os


class BatchRename:
    def __init__(self, inputpath, outputpath):
        self.inputpath = inputpath
        self.outputpath = outputpath
        self.prefix = "/all_img_"
        self.imglist = []
        self.total_img = 0
        self.strlen = 0

    def get_imgs(self):
        for root, dirs, files in os.walk(self.inputpath):
            for filename in files:
                if filename.endswith(('.png', '.jpg', '.jpeg', '.tif')):
                    self.imglist.append(root+"/"+ filename)
        self.total_img = len(self.imglist)
        self.strlen = len(str(self.total_img))
        print(f"Total image is: {self.total_img}")

    def rename(self):
        self.get_imgs()
        if 0 == self.total_img:
            return 
        for i, item in enumerate(self.imglist):
            n = self.strlen - len(str(i))
            dst = self.outputpath + self.prefix + str(0) * n + str(i) + '.jpg'
            try:
                os.rename(item, dst)
                print('converting %s ---> %s' % (item, dst))
            except Exception as e:
                print(e)


if __name__ == '__main__':
    inputpath = "D:/dataset/all_img"
    outputpath = "D:/dataset/all_img_rename"
    BatchRename(inputpath, outputpath).rename()