[ Foro de Python ]

Cómo ejecutar un código para una carpeta de imágenes

28-Mar-2022 21:39
Invitado (bapa)
0 Respuestas

Estoy creando un programa que extrae descriptores y hace una matriz de coocurrencia de las imágenes, ya que debo utilizar esos datos para entrenar una red neuronal.
Tengo el código que extrae valores de una imagen, pero no sé como hacer que corra y se ejecute para cada una de las imágenes de la carpeta sin tenerlo que hacer manualmente una por una

Este es mi código

import numpy as np
from skimage.feature import greycomatrix, greycoprops
from skimage import io, color, img_as_ubyte

img = io.imread('C:/Users/atena/Desktop/proyecto de minería/Descriptores/DB/Fire/fire 1.png')

gray = color.rgb2gray(img)
image = img_as_ubyte(gray)
io.imshow(image)

bins = np.array([0, 16, 32, 48, 64, 80, 96, 112, 128, 144, 160, 176, 192, 208, 224, 240, 255])  # 16-bit
inds = np.digitize(image, bins)

max_value = inds.max() + 1
matrix_coocurrence = greycomatrix(inds, [1], [0, np.pi / 4, np.pi / 2, 3 * np.pi / 4], levels=max_value, normed=False,
                                 symmetric=False)


# GLCM properties
def contrast_feature(matrix_coocurrence):
   contrast = greycoprops(matrix_coocurrence, 'contrast')
   return "Contrast = ", contrast


def dissimilarity_feature(matrix_coocurrence):
   dissimilarity = greycoprops(matrix_coocurrence, 'dissimilarity')
   return "Dissimilarity = ", dissimilarity


def homogeneity_feature(matrix_coocurrence):
   homogeneity = greycoprops(matrix_coocurrence, 'homogeneity')
   return "Homogeneity = ", homogeneity


def energy_feature(matrix_coocurrence):
   energy = greycoprops(matrix_coocurrence, 'energy')
   return "Energy = ", energy


def correlation_feature(matrix_coocurrence):
   correlation = greycoprops(matrix_coocurrence, 'correlation')
   return "Correlation = ", correlation


def asm_feature(matrix_coocurrence):
   asm = greycoprops(matrix_coocurrence, 'ASM')
   return "ASM = ", asm


print(contrast_feature(matrix_coocurrence))
print(dissimilarity_feature(matrix_coocurrence))
print(homogeneity_feature(matrix_coocurrence))
print(energy_feature(matrix_coocurrence))
print(correlation_feature(matrix_coocurrence))
print(asm_feature(matrix_coocurrence))




(No se puede continuar esta discusión porque tiene más de dos meses de antigüedad. Si tienes dudas parecidas, abre un nuevo hilo.)