Блог

В настоящее время у меня есть модель CNN, которую я разработал в Pytorch. Я использовал пакет hiddenlayer для создания изображения, подобного показанному на рисунке 1. Но я хочу создать изображение модели, которое должно выглядеть как-то так, как показано на рисунке 2. Есть ли какой-нибудь пакет, который я могу использовать для достижения этой цели?

изображение 1:

изображение 2:

Я лично использую figma.com и «нарисовать» ее самостоятельно, но если вы хотите создать ее автоматически, вам следует проверить этот репозиторий github, возможно, вы найдете хороший инструмент.