python 添加caffe训练数据标签的代码

如果是linux下面使用 

#!/usr/bin/python
# coding: utf-8

# -*-coding:utf-8-*-
import os
import numpy as np
from matplotlib import pyplot as plt
import shutil


#扫描文件
def GetFileList(FindPath,FlagStr=[]):
    FileList=[]
    FileNames=os.listdir(FindPath)
    if len(FileNames)>0:
        for fn in FileNames:
            if len(FlagStr)>0:
                if IsSubString(FlagStr,fn):
                    fullfilename=os.path.join(FindPath,fn)
                    FileList.append(fullfilename)
            else:
                fullfilename=os.path.join(FindPath,fn)
                FileList.append(fullfilename)
    if len(FileList)>0:
        FileList.sort()

    return FileList
def IsSubString(SubStrList,Str):
    flag=True
    for substr in SubStrList:
        if not(substr in Str):
            flag=False

    return flag

txt=open('val.txt','w')
#制作标签数据，如果是head的，标签设置为1，如果是tail的标签为１
imgfile=GetFileList('D:/xm/video/dataset/my/val/head')
for img in imgfile:
    str=img+'\t'+'1'+'\n'
    txt.writelines(str)

imgfile=GetFileList('D:/xm/video/dataset/my/val/tail')
for img in imgfile:
    str=img+'\t'+'0'+'\n'
    txt.writelines(str)
txt.close()