From b4aac24dfa9108421ceb39b81e6f07fd1c945fbc Mon Sep 17 00:00:00 2001 From: Alex Date: Sun, 6 Mar 2022 10:44:35 +0100 Subject: [PATCH] update example in docs --- docs/source/usage.rst | 4 +--- requirements.txt | 1 + setup.cfg | 4 +++- setup.py | 4 ++++ 4 files changed, 9 insertions(+), 4 deletions(-) create mode 100644 setup.py diff --git a/docs/source/usage.rst b/docs/source/usage.rst index 67c2daf..262ec1f 100644 --- a/docs/source/usage.rst +++ b/docs/source/usage.rst @@ -24,9 +24,7 @@ The following provides an example of how Luna can be used for feature visualizat color_aug = False opt_param = featurevis.OptimizationParameters(iterations, learning_rate) - aug_param = featurevis.AugmentationParameters(blur, scale, pad_crop, flip, rotation, noise, color_aug) - loss, image = featurevis.visualize_filter(image, model, "mixed5", 30, opt_param, aug_param) + loss, image = featurevis.visualize_filter(image, model, "mixed5", 30, opt_param) images.save_image(image, name="test") image_reader.save_npy_as_png("test.npy", ".") - diff --git a/requirements.txt b/requirements.txt index 47cf2b9..cd8815f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,3 +7,4 @@ m2r2 build twine matplotlib +setuptools diff --git a/setup.cfg b/setup.cfg index dcfecba..25a68e8 100644 --- a/setup.cfg +++ b/setup.cfg @@ -1,7 +1,7 @@ [metadata] # replace with your username: name = luna-fviz -version = 0.2 +version = 0.2.1 author = Alex Bäuerle author_email = alex.baeuerle@uni-ulm.de description = Feature Visualization with Tensorflow 2 @@ -21,3 +21,5 @@ install_requires = tf_slim keras numpy + matplotlib + tensorflow_addons diff --git a/setup.py b/setup.py new file mode 100644 index 0000000..7f1a176 --- /dev/null +++ b/setup.py @@ -0,0 +1,4 @@ +from setuptools import setup + +if __name__ == "__main__": + setup()