update example in docs

Sparkier · Mar 6, 2022 · b4aac24 · b4aac24
1 parent 9707d71
commit b4aac24
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 4 deletions.
diff --git 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
@@ -7,3 +7,4 @@ m2r2
 build
 twine
 matplotlib
+setuptools
diff --git 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
@@ -0,0 +1,4 @@
+from setuptools import setup
+
+if __name__ == "__main__":
+    setup()