Merge pull request #2 from johelli/master

Small cleanup to bird dataset generation script.
SandiaMLMiniApps · Feb 24, 2020 · c4d08de · c4d08de
2 parents 2050d75 + bfd467a
commit c4d08de
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 12 deletions.
diff --git a/data/generate_bird_images.py b/data/generate_bird_images.py
@@ -5,21 +5,12 @@
 import argparse
 import matplotlib.pylab as plt
 
+# Num pixels in X/Y dim for bird stamp
 bird_dim_x = 23
 bird_dim_y = 20
+# Magnitude of stamp
 bird_weight = 10.0
 
-
-
-
-
-size_x = 64;
-#size_x = 4096;
-size_y = size_x;
-
-channels_last = False
-
-
 parser = argparse.ArgumentParser(description="Generate Sandia Bird Dataset")
 
 parser.add_argument('--dim-mode', type=int, default=2, metavar='N',
@@ -86,7 +77,6 @@
 
 
 
-
 # MAIN
 def main():
 
@@ -263,5 +253,6 @@ def create_bird_stamp(rez):
 
     return bird_stamp
 
+
 if __name__ == "__main__":
     main()
diff --git a/setup_python_env.sh b/setup_python_env.sh
@@ -13,6 +13,7 @@ source minigan_torch_env/bin/activate
 
 ${PIP} install numpy 
 ${PIP} install torch
+${PIP} install torchvision
 ${PIP} install horovod
 ${PIP} install tensorboard
 ${PIP} install matplotlib==3.0.0