ComputerVisionPrograms/booleanArray.py at master · ciyengar3/ComputerVisionPrograms · 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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
import cv2
import numpy as np
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt

# initialize the list of reference points and boolean indicating
# whether cropping is being performed or not
refPt = []
cropping = False
higherbound = 5
stepsize = 10

def click_and_crop(event, x, y, flags, param):
	# grab references to the global variables
	global refPt, cropping

	# if the left mouse button was clicked, record the starting
	# (x, y) coordinates and indicate that cropping is being
	# performed
	if event == cv2.EVENT_LBUTTONDOWN:
		refPt = [(x, y)]
		cropping = True

	# check to see if the left mouse button was released
	elif event == cv2.EVENT_LBUTTONUP:
		# record the ending (x, y) coordinates and indicate that
		# the cropping operation is finished
		refPt.append((x, y))
		cropping = False

		# draw a rectangle around the region of interest
		cv2.rectangle(image, refPt[0], refPt[1], (0, 255, 0), 2)
		cv2.imshow("image", image)
# load the image, clone it, and setup the mouse callback function
image = cv2.imread("fig.png", 0)
cv2.imshow("fig", image)
clone = image.copy()
r, c = np.shape(image)
#print(r, c)
cv2.namedWindow("image")
cv2.setMouseCallback("image", click_and_crop)

# keep looping until the 'q' key is pressed
while True:
	# display the image and wait for a keypress
	cv2.imshow("image", image)
	key = cv2.waitKey(1) & 0xFF

	# if the 'r' key is pressed, reset the cropping region
	if key == ord("r"):
		image = clone.copy()

	# if the 'c' key is pressed, break from the loop
	elif key == ord("c"):
		break

# if there are two reference points, then crop the region of interest
# from teh image and display it
if len(refPt) == 2:
	roi = clone[refPt[0][1]:refPt[1][1], refPt[0][0]:refPt[1][0]]
	cv2.waitKey(0)

# close all open windows
cv2.destroyAllWindows()
#I = cv2.resize(roi, (0,0), fx=0.25, fy=0.25)

x1 = refPt[0][0]
x2 = refPt[1][0]
y1 = refPt[0][1]
y2 = refPt[1][1]
#print(refPt )
if (x2 > x1):
    higherx = x2
    lowerx = x1
else:
    lowerx = x2
    higherx = x1

if (y2 > y1):
    highery = y2
    lowery = y1
else:
    highery = y1
    lowery = y2
#print(lowerx, higherx, lowery, highery)
R = np.array([[1], [2], [3]])
counter = 0
total = 0
for x in range (lowerx, higherx + 1):
    for y in range (lowery, highery + 1):
        z = image[y, x]
        total += z
        counter += 1
        R = np.concatenate((R, np.array([[y],[x], [z]])), axis = 1)
X = R[:, 1:c]
avgValue = total / counter
print(avgValue)
r, c = np.shape(image)
booleanArray = np.zeros((r, c))
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
for y in range (0, r):
    for x in range (0, c):
        if (image[y, x] >= avgValue):
            booleanArray[y, x] = 1
for y in range (0, r, stepsize):
    for x in range (0, c, stepsize):
        if (booleanArray[y, x] == 1):
            ax.scatter(x, y, image[y,x])
ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_zlabel('Z Label')

plt.show()