分类目录结构转csv结构

分类网络目录结构转pandas csv结构:

  

   

    

     

    
    

     

      import os
     
    
   

    

     

    
    

     

      import time
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import pandas as pd
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      if __name__ == '__main__':
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          dir_path=r'F:\project\huajie\abnormal_img\dataset\abnormal_data\val/'
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    if os.path.basename(os.path.dirname(dir_path))=="train":
     
    
   

    

     

    
    

     

              val_type = 1
     
    
   

    

     

    
    

     
    elif os.path.basename(os.path.dirname(dir_path))=="val":
     
    
   

    

     

    
    

     

              val_type = 0
     
    
   

    

     

    
    

     
    else:
     
    
   

    

     

    
    

     
        print("val/train dir is error",os.path.dirname(dir_path))
     
    
   

    

     

    
    

     

              exit(1123)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          imgs = ['%s/%s' % (i[0], j) for i in os.walk(dir_path) for j in i[-1] if j.endswith(('.jpg', '.jpeg'))]
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    print("data_len",len(imgs))
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    #多分类用/分开
     
    
   

    

     

    
    

     

          columns=["img_path","cls_0/cls_1","val/train"]
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          cla_num=10
     
    
   

    

     

    
    

     

          ids=[str(i) for i in range(cla_num)]
     
    
   

    

     

    
    

     
    # ids = ["0", "1","2","3","4","","6","7"]
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          datas=[]
     
    
   

    

     

    
    

     

          start=time.time()
     
    
   

    

     

    
    

     
    for img_path in imgs:
     
    
   

    

     

    
    

     

              label_txt = os.path.basename(os.path.dirname(img_path))
     
    
   

    

     

    
    

     
        id = ids.index(label_txt)
     
    
   

    

     

    
    

     

              data_row = pd.Series([img_path.replace(dir_path,"/"), id, val_type], index=columns)
     
    
   

    

     

    
    

     

              datas.append(data_row)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          gdp4 =pd.concat(datas ,axis=1).T
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          gdp4.to_csv(dir_path+"/label.csv",index=False)
     
    
   

    

     

    
    

     
    print("time" ,time.time( ) -start,"s")
     
    
  
 

文章来源: blog.csdn.net，作者：AI视觉网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/126801233