-
Notifications
You must be signed in to change notification settings - Fork 3
/
run.py
75 lines (57 loc) · 1.56 KB
/
run.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
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
import tensorflow as tf
import numpy as np
from scipy import ndimage
import apriori
import cv2
import sys
from misc import find_item, bbox_plot
from model import vgg16
im_dir = sys.argv[1]
features, pred_vec = vgg16(im_dir)
td = {}
d = {}
test = {}
lis = []
j = 0
for feat in features:
for i in range(512):
td[j, i], d[j, i] = find_item(feat[:, :, i])
j += 1
feature_map = tf.concat((features[0, :, :, :], features[1, :, :]), axis=2)
li = [i for i in td.values() if i]
pixel = {}
for i in range(len(li)):
pixel[i] = li[i]
freq = apriori.run(pixel, 20)
# print (freq)
mk = [eval(i) for i in freq]
support = np.zeros([14, 14])
for i in range(14):
for j in range(14):
if [i, j] in mk:
support[i, j] = np.array(feature_map[i, j, :]).sum()/1024
# last_conv_output = np.squeeze(feature_map)
# import scipy
# pred = np.argmax(pred_vec)
# mat_for_mult = scipy.ndimage.zoom(last_conv_output, (16, 16, 1), order=1)
img = np.zeros([14, 14])
for i in range(14):
for j in range(14):
if [i, j] in mk:
img[i, j] = 1
new_img = np.expand_dims(img, axis=2)
new_img = tf.image.resize(new_img, (224, 224))
new = np.array(new_img[:, :, 0]).astype(int)
labeled_image, num_features = ndimage.label(new)
objs = ndimage.find_objects(labeled_image)
box = np.zeros([len(objs), 4])
# print (objs)
for i in range(len(objs)):
x = objs[i][1].start
y = objs[i][0].stop
w = objs[i][1].stop - objs[i][1].start
h = objs[i][0].start - objs[i][0].stop
box[i][:] = x, y, w, h
img = cv2.imread(im_dir, cv2.IMREAD_UNCHANGED)
org_img = cv2.resize(img, (224, 224))
bbox_plot(org_img, box, support)