def LoadData(path1, path2):
faceData = ReadDataSet(path1)
quantizer = Quantization(settings.NUM_CUTOFF)
cutoffs = quantizer.initQuantization(faceData)
quantizedFaceData = quantizer.runQuantization(faceData)
del faceData
nonfaceData = ReadDataSet(path2)
quantizedNonFaceData = quantizer.runQuantization(nonfaceData)
del nonfaceData
return quantizedFaceData, quantizedNonFaceData, cutoffs
def LoadData(path1, path2, model):
quantizer = Quantization(model.getNumCutOffs())
quantizer.setCutOffs(model.getCutOffs())
faceData = ReadDataSet(path1)
quantizedFaceData = quantizer.runQuantization(faceData)
del faceData
nonfaceData = ReadDataSet(path2)
quantizedNonFaceData = quantizer.runQuantization(nonfaceData)
del nonfaceData
return quantizedFaceData, quantizedNonFaceData
import DCT
import Quantization
import AC
import DC
import Compress
import cv2
def printBlock(block):
for row in block:
print(row)
img = cv2.imread("../photograph.jpg")
DCT = DCT.DCT()
Quantization = Quantization.Quantization()
AC = AC.AC()
DC = DC.DC()
Compress = Compress.Compress()
Y, U, V = RGB2YUV.rgb2yuv(img, img.shape[1], img.shape[0])
Y = DCT.fill(Y)
blocks = DCT.split(Y)
first = blocks[0]
print('The first block of Y:')
printBlock(first)
print('')
print('The DCT of the block:')
first = DCT.FDCT(first)
printBlock(first)
print('')
print('The Quantization of the DCT:')
