/
mosaicTest.py
50 lines (40 loc) · 1.51 KB
/
mosaicTest.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
from image_processing import process_image
from webmakeup import get_image_from_url
import glob
import os
#RefImagePath = "testing/image.jpeg"
RefImagePath = "testing/ericProf.jpg"
CpntImageDir = "testing/ericPhotos"
FileFormats = ["jpg", "jpeg", "gif", "bmp", "png"]
MosaicFilename = "eric"
def LoadRefImage():
return open(RefImagePath, "rb").read()
def LoadCpntImages():
cpntImageStreams = []
filenames = []
for fmt in FileFormats:
filenames.extend(glob.glob(CpntImageDir+'/*.'+fmt))
for file in filenames:
cpntImageStreams.append(open(file, "rb").read())
return cpntImageStreams
def Test():
refImageStream = LoadRefImage()
cpntImageStreams = LoadCpntImages()
mosaic = process_image(refImageStream, cpntImageStreams)
outFile = open(MosaicFilename+".jpg", "wb")
outFile.write(mosaic)
outFile.close()
def FacebookTest():
refImageStream = get_image_from_url(r'http://blogs-images.forbes.com/jonbruner/files/2011/07/facebook_logo.jpg')
cpntImageStreams = LoadCpntImages()
mosaic = process_image(refImageStream, cpntImageStreams)
outFile = open("facebook.jpg", "wb")
outFile.write(mosaic)
outFile.close()
def HackTest():
refImageStream = get_image_from_url(r'https://fbcdn-profile-a.akamaihd.net/hprofile-ak-snc6/211085_364348883660343_553562280_n.jpg')
cpntImageStreams = LoadCpntImages()
mosaic = process_image(refImageStream, cpntImageStreams)
outFile = open("hack.jpg", "wb")
outFile.write(mosaic)
outFile.close()