如何根据我的报告模型能力绘制或可视化U-Net？

Question

我正在准备一个演示文稿，在这里我想展示我使用的U-Net。为此，我不能只是从任何来源复制/粘贴U-Net体系结构。我必须根据模型的容量（卷积层数）进行绘制。

有没有简单的方法可以做到这一点？

我正在使用pytorch。