溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
這篇文章主要講解了“Tensorboard可視化SSD網絡結構怎么實現”，文中的講解內容簡單清晰，易于學習與理解，下面請大家跟著小編的思路慢慢深入，一起來研究和學習“Tensorboard可視化SSD網絡結構怎么實現”吧！
from torch.utils.tensorboard import SummaryWriterfrom torch.utils.tensorboard import SummaryWriterimport torchimport torchvisionfrom torchvision import datasets,transformsfrom torch.autograd import Variablefrom ssd import SSD ssd = SSD()model = ssd.net.module writer = 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()
運行結果截圖展示:
感謝各位的閱讀，以上就是“Tensorboard可視化SSD網絡結構怎么實現”的內容了，經過本文的學習后，相信大家對Tensorboard可視化SSD網絡結構怎么實現這一問題有了更深刻的體會，具體使用情況還需要大家實踐驗證。這里是億速云，小編將為大家推送更多相關知識點的文章，歡迎關注！
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
