Caffe学习笔记3

#<pre name="code"class="python">
#coding:utf-8
‘‘‘‘‘ 
Created on Jul 29, 2016 
 
@author: sgg 
‘‘‘  
  
"<span style=""font-family:Arial;font-size:18px;"">"  
"<span style=""font-size:18px;"">"  
"<span style=""font-size:18px;"">"   
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‘) 
#制作标签数据，如果是狗的，标签设置为0，如果是猫的标签为1  
imgfile=GetFileList(‘train/train_dinosar‘)#将数据集放在与.py文件相同目录下  
for img in imgfile:  
    str1=img+‘ ‘+‘4‘+‘\n‘        #用空格代替转义字符 \t   
    train_txt.writelines(str1)  
      
  
imgfile=GetFileList(‘train/train_ele‘)  
for img in imgfile:  
    str2=img+‘ ‘+‘3‘+‘\n‘  
    train_txt.writelines(str2)  


#imgfile=GetFileList(‘train/train_flower‘)#将数据集放在与.py文件相同目录下  
#for img in imgfile:  
#   str3=img+‘ ‘+‘2‘+‘\n‘        #用空格代替转义字符 \t   
#  train_txt.writelines(str3)  
      
  
#imgfile=GetFileList(‘train/train_horse‘)  
#for img in imgfile:  
#    str4=img+‘ ‘+‘1‘+‘\n‘  
#    train_txt.writelines(str4)  
  
  
#imgfile=GetFileList(‘train/train_truck‘)  
#for img in imgfile:  
#    str5=img+‘ ‘+‘0‘+‘\n‘  
#    train_txt.writelines(str5)  
train_txt.close() 

#测试集文件列表  
test_txt=open(‘val.txt‘,‘w‘)  
#制作标签数据，如果是男的，标签设置为0，如果是女的标签为1  
imgfile=GetFileList(‘val/val_dinosar‘)#将数据集放在与.py文件相同目录下  
for img in imgfile:  
    str6=img+‘ ‘+‘4‘+‘\n‘         
    test_txt.writelines(str6)  
      
  
imgfile=GetFileList(‘val/val_ele‘)  
for img in imgfile:  
    str7=img+‘ ‘+‘3‘+‘\n‘  
    test_txt.writelines(str7)  


#imgfile=GetFileList(‘val/val_flower‘)#将数据集放在与.py文件相同目录下  
#for img in imgfile:  
#    str8=img+‘ ‘+‘2‘+‘\n‘        
#    test_txt.writelines(str8)  
      
  
#imgfile=GetFileList(‘val/val_horse‘)  
#for img in imgfile:  
#    str9=img+‘ ‘+‘1‘+‘\n‘  
#   test_txt.writelines(str9)    
  
  
#imgfile=GetFileList(‘val/val_truck‘)  
#for img in imgfile:  
#   str10=img+‘ ‘+‘0‘+‘\n‘  
#    test_txt.writelines(str10)  
test_txt.close() 
  
print("成功生成文件列表")  

GLOG_logtostderr=1 $TOOLS/convert_imageset     --resize_height=$RESIZE_HEIGHT     --resize_width=$RESIZE_WIDTH     --shuffle     $TRAIN_DATA_ROOT     $DATA/train.txt     $EXAMPLE/ilsvrc12_train_lmdb

echo "Creating val lmdb..."

GLOG_logtostderr=1 $TOOLS/convert_imageset     --resize_height=$RESIZE_HEIGHT     --resize_width=$RESIZE_WIDTH     --shuffle     $VAL_DATA_ROOT     $DATA/val.txt     $EXAMPLE/ilsvrc12_val_lmdb

net: "myself/train_val.prototxt" ##模型所在目录

snapshot_prefix: "myself/caffenet_train"##生成的模型参数