search

HarisIqbal88 / PlotNeuralNet

Latex code for making neural networks diagrams
MIT License
21.74k stars 2.84k forks source link

Unet to DenseNets #50

Closed zuxfoucault closed 4 years ago

zuxfoucault commented 5 years ago

I am trying to modify Unet tex code to create DenseNets block. f838717a-6ad1-11e6-9391-f0906c80bc1d

I am wondering how to modify the connection lines that then look like densenets style? Thank you for your time!

zuxfoucault commented 5 years ago

I modified HED and made it look a bit similar to DenseNet Block. https://github.com/zuxfoucault/PlotNeuralNet/blob/master/examples/densenetblock/densenet.tex

Banyueqin commented 5 years ago

@zuxfoucault Can you tell me the python code abount the HED.Thanks.

jzhang55995 commented 4 years ago

I modified HED and made it look a bit similar to DenseNet Block. https://github.com/zuxfoucault/PlotNeuralNet/blob/master/examples/densenetblock/densenet.tex

Would you mind to share the setting for "\ConnColorA" "\ConnColorB" ... "\ConnColorD" , and "\midarrowA" ... "\midarrowC" in your source file? I couldn't find those settings in the whole package. Thanks

zuxfoucault commented 4 years ago

@jzhang55995 sorry for the late reply, here are the settings https://github.com/zuxfoucault/PlotNeuralNet/blob/master/layers/init.tex#L6

@Banyueqin sorry, I didn't modify the python part.