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
|
"""
Shell for running Local Laplacian.
"""
from local_laplacian import local_laplacian
from local_laplacian_Mullapudi2016 import local_laplacian_Mullapudi2016
import halide.imageio
import numpy as np
import sys
import timeit
def main():
if len(sys.argv) < 6:
print(f"Usage: {sys.argv[0]} input.png input.png levels alpha beta output.png")
print(f"e.g. {sys.argv[0]} input.png 8 1 1 output.png 10")
sys.exit(1)
input_path = sys.argv[1]
levels = int(sys.argv[2])
alpha = float(sys.argv[3])
beta = float(sys.argv[4])
output_path = sys.argv[5]
timing_iterations = 10
print(f"Reading from {input_path} ...")
input_buf_u8 = halide.imageio.imread(input_path)
assert input_buf_u8.dtype == np.uint8
# Convert to uint16 in range [0..1]
input_buf = input_buf_u8.astype(np.uint16) * 257
h = input_buf.shape[1]
w = input_buf.shape[2]
output_buf = np.empty([3, h, w], dtype=input_buf.dtype)
tests = {
"Manual": local_laplacian,
"Mullapudi2016": local_laplacian_Mullapudi2016,
}
for name, fn in tests.items():
print(f"Running {name}... ", end="")
t = timeit.Timer(
lambda: fn(input_buf, levels, alpha / (levels - 1), beta, output_buf)
)
avg_time_sec = t.timeit(number=timing_iterations) / timing_iterations
print("time: %fms" % (avg_time_sec * 1e3))
output_buf_u8 = (output_buf // 257).astype(np.uint8)
print(f"Saving to {output_path} ...")
halide.imageio.imwrite(output_path, output_buf_u8)
print("Success!")
sys.exit(0)
if __name__ == "__main__":
main()
|
