create a python package

.gitignore

@@ -1,3 +1,10 @@
+.idea/
+*.pyc
+dist/
+build/
+plotnn.egg-info/
+pyexamples/*.pdf
+
 *.aux
 *.log
 *.gz

MANIFEST.in

@@ -0,0 +1 @@
+include latexnn/static/*

README.md

@@ -9,43 +9,45 @@ Latex code for drawing neural networks for reports and presentation. Have a look
 - [ ] Add easy legend functionality
 - [ ] Add more layer shapes like TruncatedPyramid, 2DSheet etc
 - [ ] Add examples for RNN and likes.
+- [ ] Auto offset and size calculation
+
 
 ## Latex Usage
     see examples
 
-## PyUsage
+## Python usage
+
+### Installation
+
+Only support Python3.
+
+    python3 setup.py install
+
+### Write NN layout
 
-    mkdir my_project
-    cd my_project
-    vim my_arch.py
+The network layout looks like
 
-        import sys
-        sys.path.append('../')
-        from pycore.tikzeng import *
+    from latexnn import *
 
-        # defined your arch
-        arch = [
-            to_head( '..' ),
-            to_cor(),
-            to_begin(),
+    # defined your arch
+    arch = [
             to_Conv("conv1", 512, 64, offset="(0,0,0)", to="(0,0,0)", height=64, depth=64, width=2 ),
             to_Pool("pool1", offset="(0,0,0)", to="(conv1-east)"),
             to_Conv("conv2", 128, 64, offset="(1,0,0)", to="(pool1-east)", height=32, depth=32, width=2 ),
             to_connection( "pool1", "conv2"),
             to_Pool("pool2", offset="(0,0,0)", to="(conv2-east)", height=28, depth=28, width=1),
             to_SoftMax("soft1", 10 ,"(3,0,0)", "(pool1-east)", caption="SOFT"  ),
             to_connection("pool2", "soft1"),
-            to_end()
             ]
 
-        def main():
-            namefile = str(sys.argv[0]).split('.')[0]
-            to_generate(arch, namefile + '.tex' )
+    def main():
+        namefile = str(sys.argv[0]).split('.')[0]
+        to_pdf(arch, namefile + '.pdf' )
 
-        if __name__ == '__main__':
-            main()
+    if __name__ == '__main__':
+        main()
 
-    bash ../tikzmake.sh my_arch
+Then run `python <your_script>.py` to generate `<nn>.pdf` (replace `<nn>` with your file name).
 
 ## Examples
 

examples/HED/HED.tex

@@ -1,6 +1,6 @@
 \documentclass[border=15pt, multi, tikz]{standalone}
 \usepackage{import}
-\subimport{../../layers/}{init}
+\subimport{../../static/}{init}
 \usetikzlibrary{positioning}
 
 \def\ConvColor{rgb:yellow,5;red,2.5;white,5}

examples/SoftmaxLoss/SoftmaxLoss.tex

@@ -1,6 +1,6 @@
 \documentclass[border=15pt, multi, tikz]{standalone}
 \usepackage{import}
-\subimport{../../layers/}{init}
+\subimport{../../static/}{init}
 \usetikzlibrary{positioning}
 
 \newcommand{\up}{0.25}

examples/Unet/Unet.tex

@@ -1,7 +1,7 @@
 \documentclass[border=8pt, multi, tikz]{standalone}
 %\usepackage{blocks}
 \usepackage{import}
-\subimport{../../layers/}{init}
+\subimport{../../static/}{init}
 \usetikzlibrary{positioning}
 
 \def\ConvColor{rgb:yellow,5;red,2.5;white,5}

examples/Unet_Ushape/Unet_ushape.tex

@@ -1,7 +1,7 @@
 \documentclass[border=8pt, multi, tikz]{standalone}
 %\usepackage{blocks}
 \usepackage{import}
-\subimport{../../layers/}{init}
+\subimport{../../static/}{init}
 \usetikzlibrary{positioning}
 
 \def\ConvColor{rgb:yellow,5;red,2.5;white,5}

examples/VGG16/vgg16.tex

@@ -1,7 +1,7 @@
 \documentclass[border=15pt, multi, tikz]{standalone}
 %\usepackage{blocks}
 \usepackage{import}
-\subimport{../../layers/}{init}
+\subimport{../../static/}{init}
 \usetikzlibrary{positioning}
 
 \def\ConvColor{rgb:yellow,5;red,2.5;white,5}

examples/fcn32s/fcn32.tex

@@ -1,6 +1,6 @@
 \documentclass[border=15pt, multi, tikz]{standalone}
 \usepackage{import}
-\subimport{../../layers/}{init}
+\subimport{../../static/}{init}
 \usetikzlibrary{positioning}
 
 \def\ConvColor{rgb:yellow,5;red,2.5;white,5}

examples/fcn8s/fcn8.tex

@@ -1,6 +1,6 @@
 \documentclass[border=15pt, multi, tikz]{standalone}
 \usepackage{import}
-\subimport{../../layers/}{init}
+\subimport{../../static/}{init}
 \usetikzlibrary{positioning}
 \usetikzlibrary{3d} %for including external image 
 

latexnn/__init__.py

@@ -0,0 +1,3 @@
+from .tikzeng import *
+from .blocks import *
+from .layers import *

pycore/blocks.py → latexnn/blocks.py

@@ -1,5 +1,5 @@
 
-from .tikzeng import *
+from .layers import *
 
 #define new block
 def block_2ConvPool( name, botton, top, s_filer=256, n_filer=64, offset="(1,0,0)", size=(32,32,3.5), opacity=0.5 ):