Как ускорить поворот изображения (python, подушка) - PullRequest
Новая
       23

Как ускорить поворот изображения (python, подушка)

0 голосов
/ 09 июля 2020

Я использую следующий код для поворота изображений на 180 градусов. Размер каждого изображения составляет от 3 до 3,5 МБ. Это работает, но довольно медленно (около 1 часа для 2000 изображений).

Есть ли способ ускорить процесс? 

from PIL import Image
import pandas as pd

rootDir = "c:/Auto Picking Vision/"
imageDir = rootDir + "images/"

df = pd.read_csv(rootDir + "labels.csv")
dfSize = len(df)

for i in range(0, dfSize):
    # Open the existing file, grab the existing file's label
    fileName = df.fileName[i]
    image = Image.open(imageDir + fileName)
    label = df.label[i]
    
    # Rotate the image 180 degrees
    rotated180 = image.rotate(180)
    
    # Save the rotated image to file with a modified name
    split = fileName.split(".")
    name180 = split[0] + "-180deg." + split[1]
    rotated180.save(imageDir + name180)
    image.close()
    
    # Append the data to the dataframe
    df.loc[len(df)] = [name180, label]
...