Commit 129ddb65 authored by Martin Reinecke's avatar Martin Reinecke

more imports

parent 94f91052
import numpy as np import numpy as np
from matplotlib import pyplot as plt from matplotlib import pyplot as plt
from starblade.sugar import build_multi_starblade, multi_starblade_iteration import starblade as sb
if __name__ == '__main__': if __name__ == '__main__':
...@@ -10,10 +10,10 @@ if __name__ == '__main__': ...@@ -10,10 +10,10 @@ if __name__ == '__main__':
data = data.astype(float) data = data.astype(float)
data = data.clip(0.0001) data = data.clip(0.0001)
alpha = 1.25 alpha = 1.25
MultiStarblade = build_multi_starblade(data, alpha) MultiStarblade = sb.build_multi_starblade(data, alpha)
for i in range(10): for i in range(10):
MultiStarblade = multi_starblade_iteration(MultiStarblade, multiprocessing=True) MultiStarblade = sb.multi_starblade_iteration(MultiStarblade, multiprocessing=True)
#plotting a three channel RGB image in each iteration #plotting a three channel RGB image in each iteration
diffuse = np.empty_like(data) diffuse = np.empty_like(data)
...@@ -24,5 +24,3 @@ if __name__ == '__main__': ...@@ -24,5 +24,3 @@ if __name__ == '__main__':
plt.imsave('rgb_diffuse.jpg',diffuse/255.) plt.imsave('rgb_diffuse.jpg',diffuse/255.)
plt.imsave('rgb_point.jpg',point/255.) plt.imsave('rgb_point.jpg',point/255.)
from .sugar import build_starblade, starblade_iteration from .sugar import (build_starblade, starblade_iteration,
build_multi_starblade, multi_starblade_iteration)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment