学习总结笔记

一、数据集准备

利用该命令行，读取指定路径下图像文件的路径保存到文本文件中，用于制作训练集

ls -R  /dir/*.jpg > file.txt 

也可以通过一段python代码生成，如下所示：

    import os  
      
    root = os.getcwd() #获取当前路径  
    data = 'images'  
    path = os.listdir(root+'/'+ data) #显示该路径下所有文件  
    path.sort()  
    file = open('train.txt','w')  
      
    i = 0  
      
    for line in path:  
      str = root+'/'+ data +'/'+line  
      for child in os.listdir(str):  
        str1 = data+'/'+line+'/'+child;  
        d = ' %s' %(i)  
        t = str1 + d  
        file.write(t +'\n')  
      i=i+1  
        
    file.close()