Skip to content
Snippets Groups Projects
Commit 129ddb65 authored by Martin Reinecke's avatar Martin Reinecke
Browse files

more imports

parent 94f91052
No related branches found
No related tags found
No related merge requests found
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)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment