From 129ddb65818df318b2a5f383c0310a08dbaaff8c Mon Sep 17 00:00:00 2001 From: Martin Reinecke Date: Tue, 10 Apr 2018 15:09:41 +0200 Subject: [PATCH] more imports --- demos/multichannel_demo.py | 8 +++----- starblade/__init__.py | 3 ++- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/demos/multichannel_demo.py b/demos/multichannel_demo.py index b542ab6..efe9754 100644 --- a/demos/multichannel_demo.py +++ b/demos/multichannel_demo.py @@ -1,6 +1,6 @@ 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.) - - diff --git a/starblade/__init__.py b/starblade/__init__.py index 4986aee..f86f342 100644 --- a/starblade/__init__.py +++ b/starblade/__init__.py @@ -1 +1,2 @@ -from .sugar import build_starblade, starblade_iteration +from .sugar import (build_starblade, starblade_iteration, + build_multi_starblade, multi_starblade_iteration) -- GitLab