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

more imports

parent 94f91052
import numpy as np
from matplotlib import pyplot as plt
from starblade.sugar import build_multi_starblade, multi_starblade_iteration
import starblade as sb
if __name__ == '__main__':
......@@ -10,10 +10,10 @@ if __name__ == '__main__':
data = data.astype(float)
data = data.clip(0.0001)
alpha = 1.25
MultiStarblade = build_multi_starblade(data, alpha)
MultiStarblade = sb.build_multi_starblade(data, alpha)
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
diffuse = np.empty_like(data)
......@@ -24,5 +24,3 @@ if __name__ == '__main__':
plt.imsave('rgb_diffuse.jpg',diffuse/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