caffe:制作自己的数据集train.txt和val.txt,生成lmdb文件

一、准备图片

在caffe/路径下:新建一个文件夹images

caffe/images/data/路径下存放着自己的图片数据集:

分为train和val,里面放着所有的图片

二、生成标签文件.txt文件制作

import os  
  
def IsSubString(SubStrList,Str):  
    flag=True  
    for substr in SubStrList:  
        if not(substr in Str):  
            flag=False  
      
    return flag  
  
#扫面文件  
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  
  
  
  
train_txt=open('train.txt','w')  
#制作标签数据,如果是

猜你喜欢

转载自blog.csdn.net/yql_617540298/article/details/89819851