RamHacks2016/main.py at master · willowkap/RamHacks2016 · GitHub

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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
import numpy as np
import argparse
import imutils
import glob
import cv2
import os
import urllib2
import urllib
import re
import requests
import json


mainUrl = "https://s3.amazonaws.com/elephant-ramhacks-dev/"
postUrl = "https://ramhacks-2016.herokuapp.com/certifies"
response = urllib2.urlopen(mainUrl)
html = response.read()
root = [a for a in re.findall("<Key>(.*?)<", html)]
outputDict = {}
for i in range(len(root)-1,-1,-1):
	if root[i].endswith("/"):
		root.pop(i)


for element in root:
	target_folder, ignore = element.split("/",1)

	if not os.path.exists(target_folder):
		os.makedirs(target_folder)
	urllib.urlretrieve(mainUrl+element, element)


listOfFiles = [filename for filename in os.listdir('source') if filename.endswith('jpg')]
listOfTemplateFiles = [filename for filename in os.listdir('targets') if filename.endswith('png')]
maxFileName = ""
maxNum = 0
maxFileIndex = 0

smallImages = [cv2.imread("targets/" + filename) for filename in listOfTemplateFiles]
smallImages = map(lambda a: cv2.cvtColor(a, cv2.COLOR_BGR2GRAY), smallImages)
smallImages = map(lambda a: cv2.Canny(a, 50, 200), smallImages)


for mainImgName in  listOfFiles:
	mainImg = cv2.imread("source/" + mainImgName)
	mainImg = cv2.cvtColor(mainImg, cv2.COLOR_BGR2GRAY)
	mainImg = cv2.Canny(mainImg, 50, 200)

	for i in range(len(smallImages)):
		result = cv2.matchTemplate(mainImg, smallImages[i], cv2.TM_CCOEFF)
		(_, maxVal, _, maxLoc) = cv2.minMaxLoc(result)
		if(maxVal > maxNum):
			maxNum = maxVal
			maxFileName = listOfTemplateFiles[i]
			maxFileIndex = i
	smallImages.pop(maxFileIndex)
	print listOfTemplateFiles[maxFileIndex] + " is a subimage of " + mainImgName
	outputDict[listOfTemplateFiles[maxFileIndex][:listOfTemplateFiles[maxFileIndex].find(".")]] = mainImgName[:mainImgName.find(".")]
	listOfTemplateFiles.pop(maxFileIndex)
	maxNum = 0
	maxFileIndex = 0

print outputDict

payload = {	'env': 'dev',
			'answers': outputDict}
headers = {'Content-Type': 'application/json'}

response = requests.post(postUrl, data=json.dumps(payload), headers=headers)

for element in response:
	print element