Thesis-NoiseDetection-CNN/modules/image_metrics/svd_metric.py

# module file which contains all image metrics used in project

from numpy.linalg import svd
from PIL import Image
from scipy import misc

import time
import numpy as np
from sklearn import preprocessing

'''
Method which extracts SVD features from image and returns 's' vector
@return 's' vector
'''
def get_s_model_data(image):
    U, s, V = svd(image, full_matrices=False)
    size = len(s)

    # normalized output
    output_normalized = preprocessing.normalize(s, norm='l2', axis=1, copy=True, return_norm=False)

    result = output_normalized.reshape([size, 1, 3])
    return result

def get(image):
    return svd(image, full_matrices=False)

def get_s(image):
    U, s, V = svd(image, full_matrices=False)
    return s

def get_U(image):
    U, s, V = svd(image, full_matrices=False)
    return U

def get_V(image):
    U, s, V = svd(image, full_matrices=False)
    return V