根據csv中的影像標簽創建子檔案夾

我想為每個類創建帶有相關影像的子檔案夾。現在，我只有一個包含所有影像的檔案夾。我創建了一個帶有 images_id 和標簽的 .csv。我能夠根據標簽創建子檔案夾，但無法移動相關子檔案夾中的影像。

我的檔案夾結構如下所示：

• 資料（影像）
  • 圖片1.jpg
  • img2.jpg
  • ....jpg
• 標簽.csv（image_id，標簽）
• 筆記本.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)

uj5u.com熱心網友回復：

我相信這是這部分：

#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)

建議

• 只需對您的類名進行一次回圈（您回圈了兩次）
• 將目標路徑存盤到變數

代碼：

#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)

標籤：Python 熊猫 图片 目录 标签

上一篇：什么是Java中的密封類17

下一篇：從Edit中獲取價值以處理區域