-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
26 lines (22 loc) · 917 Bytes
/
main.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
import numpy as np
import util
import matplotlib
from matplotlib import pyplot as plt
import visual_words
import visual_recog
import skimage.io
if __name__ == '__main__':
num_cores = util.get_num_CPU()
path_img = "../data/waterfall/sun_bolfhwtizbvyjmem.jpg"
image = skimage.io.imread(path_img)
image = image.astype('float')/255
filter_responses = visual_words.extract_filter_responses(image)
util.display_filter_responses(filter_responses)
visual_words.compute_dictionary(num_workers=num_cores)
dictionary = np.load('dictionary.npy')
wordmap = visual_words.get_visual_words(image,dictionary)
util.save_wordmap(wordmap, "waterfall_3.jpg")
visual_recog.build_recognition_system(num_workers=num_cores)
conf, accurSacy = visual_recog.evaluate_recognition_system(num_workers=num_cores)
print(conf)
print(np.diag(conf).sum()/conf.sum())