How to realize the Visual SSD Network structure of Tensorboard

This article mainly explains the "Tensorboard visual SSD network structure how to achieve", the content of the article is simple and clear, easy to learn and understand, now please follow the editor's ideas slowly in depth, together to study and learn "Tensorboard visual SSD network structure how to achieve" it!

From torch.utils.tensorboard import SummaryWriterfrom torch.utils.tensorboard import SummaryWriterimport torchimport torchvisionfrom torchvision import datasets,transformsfrom torch.autograd import Variablefrom ssd import SSDssd = SSD () model= ssd.net.modulewriter = SummaryWriter () for i in range (5): images = torch.randn (4,3,300,300). Cuda () writer.add_graph (model, input_to_model=images, verbose=False) writer.flush () writer.close ()

The screenshot of the running result shows:

Thank you for reading, the above is the "Tensorboard visualization SSD network structure how to achieve" content, after the study of this article, I believe you on the Tensorboard visualization SSD network structure how to achieve this problem has a deeper understanding, the specific use of the situation also needs to be verified in practice. Here is, the editor will push for you more related knowledge points of the article, welcome to follow!