perspective-control/fixbuilding.py at main · jointpoints/perspective-control · 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
__author__ = 'Andrei Eliseev'


from sys import argv
from os import sep
import perspective_control.verbosity


if __name__ == '__main__':
	# 1. Process arguments
	helpline = 'USAGE:\n\tpython .' + sep + 'fixbuilding.py <image_path> [-ov]\nWHERE:\n\t<image_path> : a path to image of a building;\n\t-o           : show original image in final result demonstration.\n\t-v           : switch on verbosity.\n'
	verbosity_label_capacity = 51
	args = set(argv[1:])
	if len(args) == 0:
		print(helpline)
		exit()
	show_original = '-o' in args
	verbosity = '-v' in args
	args = args - {'-o', '-v'}
	if len(args) != 1:
		print(helpline)
		exit()
	image_path = tuple(args)[0]
	if verbosity:
		print(f'Perspective fix for "{image_path}".\n')
		print(' ' * 23 + 'Step' + ' ' * 23 + '|' + ' ' * 6 + 'Progress' + ' ' * 6 + '|', end='')

	# 2. Necessary imports
	perspective_control.verbosity.start_progress(verbosity, 'Initialisation', verbosity_label_capacity)
	import perspective_control.vanishing_points
	perspective_control.verbosity.update_progress(verbosity, 5)
	import perspective_control.transform
	perspective_control.verbosity.update_progress(verbosity, 5)
	from skimage import io
	perspective_control.verbosity.update_progress(verbosity, 5)
	from matplotlib import pyplot as plt
	perspective_control.verbosity.update_progress(verbosity, 5)

	# 3. Open the image
	perspective_control.verbosity.start_progress(verbosity, 'Image loading', verbosity_label_capacity)
	try:
		image = io.imread(image_path)
		perspective_control.verbosity.update_progress(verbosity, 20)
	except:
		perspective_control.verbosity.update_progress(verbosity, 20)
		print('\n\nERROR. Image was not found.')
		exit()

	# 4. Find vanishing points
	v_h, v_v = perspective_control.vanishing_points.get_vanishing_points(image, verbosity)

	# 5. Get rid of vanishing points
	corrected_image = perspective_control.transform.remove_perspective(image, v_h, v_v, verbosity)

	# 6. Show the result
	if not show_original:
		plt.imshow(corrected_image)
		plt.xticks([])
		plt.yticks([])
	else:
		plt.subplot(1, 2, 1)
		plt.imshow(image)
		plt.xticks([])
		plt.yticks([])
		plt.subplot(1, 2, 2)
		plt.imshow(corrected_image)
		plt.xticks([])
		plt.yticks([])
	plt.show()