根据 csv 中的图像标签创建子文件夹

Question

我想为每个 class 创建包含相关图像的子文件夹。现在，我只有一个包含所有图像的文件夹。我创建了一个带有 images_id 和标签的 .csv。我能够根据标签创建子文件夹，但我无法移动相关子文件夹中的图像。

我的文件夹结构如下所示：

数据（图片）
- img1.jpg
- img2.jpg
- ...jpg
labels.csv（image_id，标签）
notebook.ipynb

import shutil, os
import pandas as pd

labels = pd.read_csv("labels.csv")
labels = labels.sort_values('Manu_Series_Year')

class_names = list(labels.Manu_Series_Year.unique())

train_images = 'train'
train_cat = 'train_'

#creating subfolders
for i in class_names:
    os.makedirs(os.path.join('train_', i))

#moving the image files to their respective categories
for c in class_names: # Category Name
    for i in list(labels[labels['Manu_Series_Year']==c]['Image_id']): # Image Id
        get_image = os.path.join('train', str(i)) # Path to Images 
        move_image_to_cat = shutil.move(get_image, 'train_/'+c)

Traceback 
---------------------------------------------------------------------------
FileNotFoundError                         Traceback (most recent call last)
C:\Program Files\WindowsApps\PythonSoftwareFoundation.Python.3.9_3.9.2032.0_x64__qbz5n2kfra8p0\lib\shutil.py in move(src, dst, copy_function)
    813     try:
--> 814         os.rename(src, real_dst)
    815     except OSError:

FileNotFoundError: [WinError 3] Das System kann den angegebenen Pfad nicht finden: 'C://Users/deniz/OneDrive/MA/Dataset/archive_unbekannt/train\1' -> 'C://Users/deniz/OneDrive/MA/Dataset/archive_unbekannt/train_/Acura ILX 2013'

During handling of the above exception, another exception occurred:

FileNotFoundError                         Traceback (most recent call last)
<ipython-input-28-8d99c8eb764e> in <module>
     18     for i in list(labels[labels['Manu_Series_Year']==c]['Image_id']): # Image Id
     19         get_image = os.path.join('C://Users/deniz/OneDrive/MA/Dataset/archive_unbekannt/train', str(i)) # Path to Images
---> 20         move_image_to_cat = shutil.move(get_image, 'C://Users/deniz/OneDrive/MA/Dataset/archive_unbekannt/train_/'+c)

C:\Program Files\WindowsApps\PythonSoftwareFoundation.Python.3.9_3.9.2032.0_x64__qbz5n2kfra8p0\lib\shutil.py in move(src, dst, copy_function)
    832             rmtree(src)
    833         else:
--> 834             copy_function(src, real_dst)
    835             os.unlink(src)
    836     return real_dst

C:\Program Files\WindowsApps\PythonSoftwareFoundation.Python.3.9_3.9.2032.0_x64__qbz5n2kfra8p0\lib\shutil.py in copy2(src, dst, follow_symlinks)
    441     if os.path.isdir(dst):
    442         dst = os.path.join(dst, os.path.basename(src))
--> 443     copyfile(src, dst, follow_symlinks=follow_symlinks)
    444     copystat(src, dst, follow_symlinks=follow_symlinks)
    445     return dst

C:\Program Files\WindowsApps\PythonSoftwareFoundation.Python.3.9_3.9.2032.0_x64__qbz5n2kfra8p0\lib\shutil.py in copyfile(src, dst, follow_symlinks)
    263     else:
    264         try:
--> 265             with open(src, 'rb') as fsrc, open(dst, 'wb') as fdst:
    266                 # macOS
    267                 if _HAS_FCOPYFILE:

FileNotFoundError: [Errno 2] No such file or directory: 'train\1'

更新

import shutil, os
import pandas as pd

labels = pd.read_csv("labels.csv")
labels = labels.sort_values('Manu_Series_Year')

class_names = list(labels.Manu_Series_Year.unique())

train_images = 'data'
train_cat = 'data_'

#creating subfolders
for c in class_names:
    dest = 'data_' + c
    os.makedirs(c)
    for i in list(labels[labels['Manu_Series_Year']==c]['Image_id']): # Image Id
        get_image = os.path.join('data', str(i)) # Path to Images 
        move_image_to_cat = shutil.move(get_image, dest)

更新2

# should go in the right direction
# works
import shutil, os
import pandas as pd

# read label
labels = pd.read_csv("labels.csv")
# sort label
labels = labels.sort_values('Manu_Series_Year')

# take unique labels
class_names = list(labels.Manu_Series_Year.unique())

# loc of train data
train_images = 'data'
# loc of labels with img
train_cat = 'data_'

#creating subfolders
for c in class_names:
    dest = 'data_/' + c
    os.makedirs(dest)
    for i in list(labels[labels['Manu_Series_Year']==c]['Image_id']): # Image Id
        get_image = os.path.join('data', i) # Path to Images 
        move_image_to_cat = shutil.move(get_image, dest)

Answer 1

我相信是这部分：

#creating subfolders
for i in class_names:
    os.makedirs(os.path.join('train_', i))

创建的文件夹类似于 root/train_/[class_name]

然后你尝试移动你的图像：

move_image_to_cat = shutil.move(get_image, 'train_/'+c)

建议

只需对您的 class 名称进行一次循环（您正在循环两次）
将目标路径存储到变量

代码：

#creating subfolders
for c in class_names:
    dest = 'train_' + c
    os.makedirs(dest)
    for i in list(labels[labels['Manu_Series_Year']==c]['Image_id']): # Image Id
        get_image = os.path.join('data', i) # Path to Images 
        move_image_to_cat = shutil.move(get_image, dest)

根据 csv 中的图像标签创建子文件夹

Create subfolders bases on image labels in csv

python

directory

label

image

pandas

建议