基于ssd_mobilenet模型实现目标检测(含数据集制作)

参考链接:https://www.cnblogs.com/White-xzx/p/9503203.html

1.参考VOC数据集建立目录结构,Annotations中放置标记的XML文件,JPEGImage放置标记的图片,Tools放置一些工具
在这里插入图片描述
对JPEGImages文件夹中新增文件的批量重命名工具:

# -*- coding:utf8 -*-  
import os
import sys
class BatchRename():
    ''''' 
    批量重命名文件夹中的图片文件 
    '''
    def __init__(self):
        #给定目标目录  
        self.path = '../JPEGImages'

    def rename(self):
        filelist = os.listdir(self.path)
        total_num = len(filelist)
        print ("total_num : %d" % total_num)
        i = 1
        n = 6

        for item in filelist:
            if item.startswith('2019_'):
                i = i + 1
        print ("pass files num: %d" % (i-1))

        for item in filelist:
            if item.startswith('2019_'):
                continue
            if item.endswith('.jpg'):
                n = 6 - len(str(i))
                src = os.path.join(os.path.abspath(self.path), item)
                dst = os.path.join(os.path.abspath(self.path), "2019_"+str(0)*n + str(i) + '.jpg')
                try:
                    os.rename(src, dst)
                    print ('converting %s to %s ...' % (src, dst))
                    i = i + 1
                except:
                    continue
if __name__ == '__main__':
    demo = BatchRename()
    demo.rename()

你可能感兴趣的:(Tensorflow)