123456789101112131415161718192021222324252627282930313233343536373839 |
- # 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='l1', axis=0, 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
|