Add some network structure diagrams

[WZMIAOMIAO]

🚀 The feature

Add some network structure diagrams in torchvision repository. I want to contribute some drawn diagrams.

Motivation, pitch

In order to facilitate everyone to learn the model in the torchvison.

Alternatives

No response

Additional context

I draw the structure diagram of FCN network, reference torchvision official FCN model. I wish it can be added to the repository to facilitate everyone's study.

[datumbox]

I think that would be very interesting.

As far as I know, there is no great automated way to produce such diagrams. If there was a good way to do this, I think it would be beneficial to include them in key models or building blocks. I would love to hear ideas on that front.

[WZMIAOMIAO]

I also hope to use automated way to produce such diagrams. But I haven't found any ideal drawing tool yet. All these pictures were drawn manually by myself( using drawio tools). I simply want to share it to help more learners.

[WZMIAOMIAO]

I draw the structure diagram of DeepLabv3 network, reference torchvision official DeepLabv3_Resnet50 model.

[WZMIAOMIAO]

I draw the structure diagram of LRASPP network, reference torchvision official lraspp_mobilenetv3 model.

[prabhat00155]

We could perhaps use tools like https://github.com/szagoruyko/pytorchviz or https://github.com/lutzroeder/netron to visualise the pre-trained models.

[datumbox]

I've tried PyTorchViz, though it helps you build graphs easily, I don't like the fact that it visualizes the operations on the tensor-level than on the nn.Module level. My first impressions concerning Netron is that it's a viable solution.