Сохранение в нескольких файлах csv и загрузка изображений из набора данных - PullRequest
Новая
       43

Сохранение в нескольких файлах csv и загрузка изображений из набора данных

1 голос
/ 13 июля 2020

Я пытаюсь загрузить набор данных, содержащий 5k изображений, но получаю эту ошибку: 

image = cv2.resize(image2, (224, 224))
TypeError: Expected Ptr<cv::UMat> for argument 'src'

Но когда я помещаю только одно изображение, все работает нормально.

Последний вопрос : Возможно ли каким-то образом, когда набор данных с таким количеством загруженных изображений, сохранить их функции (из скользящего окна, которое я использую) в нескольких файлах CSV? Например: fashion1.png -> fashion1.csv, fashion2.png -> fashion2.csv, ... fashion5000.png -> fashion5000.csv? 

import cv2
import matplotlib.pyplot as plt
import numpy as np

#image2 = cv2.imread("fashion1.png")# your image path
image2 = r"C:\Users\John\Desktop\new\fashion\img\*.png"
image = cv2.resize(image2, (224, 224))
tmp = image  # for drawing a rectangle
stepSize = 20
(w_width, w_height) = (60, 60 )  # window size
for x in range(0, image.shape[1] - w_width, stepSize):
    for y in range(0, image.shape[0] - w_height, stepSize):
        window = image[x:x + w_width, y:y + w_height, :]
        cv2.rectangle(tmp, (x, y), (x + w_width, y + w_height), (255, 0, 0), 2)
        plt.imshow(np.array(tmp).astype('uint8'))
        plt.show()
        mean_values=[]
        mean_val, std_dev = cv2.meanStdDev(image)
        mean_val = mean_val[:3]
        mean_values.append([mean_val])
        mean_values = np.asarray(mean_values)
        mean_values2 = mean_values.reshape(1,3)
        result = mean_values2.flatten()
        print(result)
        for i in range(len(result)):
            result_file = open('fashion1.csv', 'a')
            result_file.write("{}{}".format(result[i], '\n'))

1 Ответ

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

Во-первых: чтобы изменить размер изображения, вы должны прочитать его в памяти - resize() не может использовать имя файла.

Второе: для чтения *.png вы должны использовать glob.glob(), чтобы все имена файлов совпадали этот шаблон и позже используйте for -l oop, чтобы читать каждое изображение отдельно, изменять его размер и даже создавать имя файла с расширением .csv

import glob
import cv2

pattern = r"C:\Users\John\Desktop\new\fashion\img\*.png"    

for filename in glob.glob(pattern):
    print('PNG:', filename)

    image = cv2.imread(filename)
    image = cv2.resize(image, (224, 224))
    
    csv_filename = filename.replace('.png', '.csv')
    print('CSV:', csv_filename)

    result = [1,2,3,4,5,6]
    
    with open(csv_filename, 'w') as fh:
        for value in result:
            fh.write('{}\n'.format(value))

Кстати: Если я понимаю код, это может быть проще 

import glob
import cv2

# --- functions ---

def process(filename):
    
    print('image:', filename)

    image = cv2.imread(filename)
    image = cv2.resize(image, (224, 224))
    
    step_size = 20
    window_width = 60
    window_height = 60
    
    width  = image.shape[1] - window_width
    height = image.shape[0] - window_height
    
    # --- get all results ---
    
    results = []
    
    for x in range(0, width, step_size):
        for y in range(0, height, step_size):
            image_temp = image.copy()  # for drawing a rectangle
    
            window = image[y:y+window_height, x:x+window_width]
            
            cv2.rectangle(image_temp, (x, y), (x+window_width, y+window_height), (255, 0, 0), 2)

            cv2.imshow('image', image_temp.astype('uint8'))
            # cv2.imshow() needs cv2.waitKey() to update image in window
            cv2.waitKey(1) # wait only 1 ms for key
            #cv2.waitKey(0) # wait until you press any key
            
            mean_val, std_dev = cv2.meanStdDev(window)
            mean_val = mean_val[:3].flatten()
            #print(mean_val)
            results.extend(mean_val)

    cv2.destroyAllWindows() # close window at the end

    # --- write all results ---

    csv_filename = filename.replace('.png', '.csv')

    print('  csv:', csv_filename)

    with open(csv_filename, 'w') as fh:
        for value in results:
            fh.write("{}\n".format(value))

    print('------')

# --- main ---            

pattern = r"C:\Users\John\Desktop\new\fashion\img\*.png"    

for filename in glob.glob(pattern):
    process(filename)
...