-
Notifications
You must be signed in to change notification settings - Fork 8
/
Copy pathResizeDemo.py
124 lines (103 loc) · 3.42 KB
/
ResizeDemo.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
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
115
116
117
118
119
120
121
122
123
import argparse, png, array
from readPNG_2 import Picture
#from SeamCarverLib import SeamCarver
import SeamCarverLib
import SeamCarverUtilities
import pdb
parser = argparse.ArgumentParser()
parser.add_argument("filename", help="filename of image", type=str)
parser.add_argument("rowsToRemove", help="number of rows to remove from image", type=int)
parser.add_argument("colsToRemove", help="number of cols to remove from image", type=int)
args = parser.parse_args()
print "build energy array for {}".format(args.filename)
pic = Picture(args.filename)
print "create seam carver"
sc = SeamCarverLib.SeamCarver(pic)
print "print vertical seam 1"
print sc.width(), sc.height()
#SeamCarverUtilities.printVerticalSeamEnergy(sc)
SeamCarverUtilities.printVerticalSeam(sc)
#SeamCarverUtilities.distToArray(sc)
print "remove vertical seam"
s = sc.findVerticalSeam()
sc.removeVerticalSeam(s)
print sc.width(), sc.height()
print "find vertical seam"
SeamCarverUtilities.printVerticalSeam(sc)
print "print horizontal seam"
SeamCarverUtilities.printHorizontalSeam(sc)
#SeamCarverUtilities.distToArray(sc)
#SeamCarverUtilities.printHorizontalSeamEnergy(sc)
print "remove horizontal seam"
s = sc.findHorizontalSeam()
sc.removeHorizontalSeam(s)
print sc.width(), sc.height()
#SeamCarverUtilities.printHorizontalSeamEnergy(sc)
print "find horizontal seam"
SeamCarverUtilities.printHorizontalSeam(sc)
#SeamCarverUtilities.distToArray(sc)
# s = sc.findVerticalSeam()
# sc.removeVerticalSeam(s)
# # print "print vertical seam 2"
# print sc.width(), sc.height()
# SeamCarverUtilities.printVerticalSeam(sc)
# #SeamCarverUtilities.printVerticalSeamEnergy(sc)
# # s = sc.findVerticalSeam()
# # for i in range(sc.height()):
# # print s[i]
# # # print sc._edgeTo
# # # print sc._distTo
# # SeamCarverUtilities.printVerticalSeam(sc)
# # print
# s = sc.findVerticalSeam()
# sc.removeVerticalSeam(s)
# print sc.width(), sc.height()
# # print "print vertical seam 3"
# # print sc.width(), sc.height()
# SeamCarverUtilities.printVerticalSeam(sc)
# SeamCarverUtilities.distToArray(sc)
# #SeamCarverUtilities.printVerticalSeamEnergy(sc)
# # s = sc.findVerticalSeam()
# # for i in range(sc.height()):
# # print s[i]
# s = sc.findVerticalSeam()
# sc.removeVerticalSeam(s)
# print sc.width(), sc.height()
# # print "print vertical seam 4"
# # print sc.width(), sc.height()
# SeamCarverUtilities.printVerticalSeam(sc)
#SeamCarverUtilities.printVerticalSeamEnergy(sc)
# s = sc.findVerticalSeam()
# for i in range(sc.height()):
# print s[i]
# print "print vertical seam 2"
# # print sc._edgeTo
# # print sc._distTo
# s = sc.findVerticalSeam()
# sc.removeVerticalSeam(s)
# print sc.height(), sc.width()
#SeamCarverUtilities.printVerticalSeam(sc)
# print
# print "print vertical seam 3"
# s = sc.findVerticalSeam()
# sc.removeVerticalSeam(s)
# print sc.height(), sc.width()
# # print sc._edgeTo
# # print sc._distTo
# #pdb.set_trace()
#SeamCarverUtilities.printVerticalSeam(sc)
# print
# print "remove vertical seam 4"
# s = sc.findVerticalSeam()
# sc.removeVerticalSeam(s)
# print sc.height(), sc.width()
# # print sc._edgeTo
# # print sc._distTo
# SeamCarverUtilities.printVerticalSeam(sc)
# print "remove vertical seam 5"
# s = sc.findVerticalSeam()
# sc.removeVerticalSeam(s)
# print sc.height(), sc.width()
# SeamCarverUtilities.printVerticalSeam(sc)
#print "remove {} rows".format(args.rowsToRemove)
#print "remove {} cols".format(args.colsToRemove)