使用python将图片按标签分入不同文件夹的方法-6ren

使用python将图片按标签分入不同文件夹的方法

转载作者：qq735679552 更新时间：2022-09-27 22:32:09

26

4

CFSDN坚持开源创造价值，我们致力于搭建一个资源共享平台，让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章使用python将图片按标签分入不同文件夹的方法由作者收集整理，如果你对这篇文章有兴趣，记得点赞哟.

给定图像集如下，所有类别的图片均在一个文件夹内:

使用python将图片按标签分入不同文件夹的方法

给定与图片名相匹配的表格，声明每张图片对应的类别（共有20个类别）:

使用python将图片按标签分入不同文件夹的方法

那么，如何根据表格中所给的类别将图片分入对应的文件夹内呢？以我的情况为例，我想将图片分为20类（category_id有0-19共20类），可利用下面的代码进行分类（经细心网友指正，代码已做出修改）.

 
    ? 
   
         #引入相关库 
        
         import 
         pandas as pd 
        
         import 
         os 
        
         import 
         shutil  
         #用于移动文件

 
    ? 
   
 
     
       
       
         #打开表格文件并读取 
        
 
         f 
         = 
         open 
         ( 
         "list.csv" 
         , 
         "rb" 
         )  
         #输入表格所在路径+名称 
        
 
         list 
         = 
         pd.read_csv(f) 
        
 
         list 
         [ 
         "file_id_jpg" 
         ] 
         = 
         ".jpg" 
         #建立图片名与类别相对应 
        
 
         list 
         [ 
         "file_id1" 
         ] 
         = 
         list 
         [ 
         "file_id" 
         ] 
         + 
         list 
         [ 
         "file_id_jpg" 
         ]  
         #建立图片名与类别相对应 
        
 
     
 
   

 
    ? 
   
         #进行分类 
        
         for 
         i  
         in 
         range 
         ( 
         0 
         , 
         20 
         ): 
        
         listnew 
         = 
         list 
         [ 
         list 
         [ 
         "category_id" 
         ] 
         = 
         = 
         i] 
        
         l 
         = 
         listnew[ 
         "file_id1" 
         ].tolist() 
        
         j 
         = 
         str 
         (i) 
        
         for 
         each  
         in 
         l: 
        
         shutil.move(each,j)