histogramGLCMAndColor_RGB.py

import os
import numpy as np
import pandas as pd
from skimage import io, color, img_as_ubyte
from skimage.feature import graycomatrix, graycoprops
from skimage.exposure import equalize_hist
from tqdm import tqdm


# Function to calculate GLCM features
def calculate_glcm(image):
    # Convert image to grayscale
    gray_image = color.rgb2gray(image)

    gray_image = equalize_hist(gray_image)

    # Convert to uint8 for GLCM computation
    gray_image = img_as_ubyte(gray_image)

    # Define GLCM properties
    distances = [1, 2, 3]  # Distances for co-occurrence matrix
    angles = [0, np.pi / 4, np.pi / 2, 3 * np.pi / 4]  # Angles for co-occurrence matrix

    # Calculate GLCM
    glcm = graycomatrix(
        gray_image, distances=distances, angles=angles, symmetric=True, normed=True
    )

    # Calculate GLCM properties
    contrast = graycoprops(glcm, "contrast")
    correlation = graycoprops(glcm, "correlation")
    dissimilarity = graycoprops(glcm, "dissimilarity")
    homogeneity = graycoprops(glcm, "homogeneity")
    energy = graycoprops(glcm, "energy")

    # Return GLCM features as a 1D array
    return np.array(
        [
            contrast.mean(),
            correlation.mean(),
            dissimilarity.mean(),
            homogeneity.mean(),
            energy.mean(),
        ]
    )


# Function to calculate average color (RGB) features
def calculate_average_color(image):
    # Calculate average RGB values
    avg_color = np.mean(image, axis=(0, 1))

    return avg_color


# Function to load images and extract features
def load_images_and_extract_features(folder):
    features = []  # Features
    labels = []  # Labels
    for class_folder in os.listdir(folder):
        if not os.path.isdir(os.path.join(folder, class_folder)):
            continue  # Skip if not a directory
        for filename in tqdm(os.listdir(os.path.join(folder, class_folder))):
            if filename.endswith(".jpg") or filename.endswith(".png"):
                image = io.imread(os.path.join(folder, class_folder, filename))
                glcm_features = calculate_glcm(image)
                color_features = calculate_average_color(image)

                # Combine GLCM and color features
                features.append(np.concatenate((glcm_features, color_features)))

                labels.append(class_folder)  # Use folder name as label

    return features, labels


# Modify the RGB data folder path
rgb_folder = "data/RGB data"

# Load images and extract features for train and test folders
train_features, train_labels = load_images_and_extract_features(
    os.path.join(rgb_folder, "train")
)
test_features, test_labels = load_images_and_extract_features(
    os.path.join(rgb_folder, "test")
)

# Create DataFrames for train and test sets
train_df = pd.DataFrame(
    train_features,
    columns=[
        "Contrast",
        "Correlation",
        "Dissimilarity",
        "Homogeneity",
        "Energy",
        "Avg_Red",
        "Avg_Green",
        "Avg_Blue",
    ],
)
train_df["Label"] = train_labels

test_df = pd.DataFrame(
    test_features,
    columns=[
        "Contrast",
        "Correlation",
        "Dissimilarity",
        "Homogeneity",
        "Energy",
        "Avg_Red",
        "Avg_Green",
        "Avg_Blue",
    ],
)
test_df["Label"] = test_labels

# Save the DataFrames to CSV files
train_csv = "train_RGB.csv"
test_csv = "test_RGB.csv"

train_df.to_csv(train_csv, index=False)
test_df.to_csv(test_csv, index=False)

print("Train CSV saved to:", train_csv)
print("Test CSV saved to:", test_csv)