IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 人工智能 -> torch网络展示 -> 正文阅读

[人工智能]torch网络展示

1、打印网络信息

import timm
import torch
from torchsummary import summary
model = timm.create_model('regnetx_002', pretrained=False,num_classes=100)##regnety_032
summary(model, ( 3, 1280,960))
timm.list_models('regnet*')

print(model)#打印的是网络每层的信息

?print(summary(model, ( 3, 1280,960)))#打印的是权重信息,参数的权重是按照float32记录的,站4个字节

?

?2、绘制网络结构图

import hiddenlayer as hl
import torch
def save_graph(model, input_size, save_name):
    model_graph = hl.build_graph(model, torch.zeros(input_size))
    model_graph.theme = hl.graph.THEMES["blue"].copy()
    model_graph.save(path="%s_graph.jpg"%save_name, format="jpg")

save_graph(model, (8, 3, 1280,960),"inception_v3")

?3、绘制网络权重

使用的是netron,输入的是pt,网上给出的还会有网络结构但是没实现

?4、利用tensorboard

import torch
import torch.nn as nn
import torch.nn.functional as F
from tensorboardX import SummaryWriter  # 用于进行可视化
from torchviz import make_dot

# modelviz = modelViz()
# 创建输入
sampledata = torch.rand(1, 3, 1280,960)
# 看看输出结果对不对
out = model(sampledata)
print(out)  # 测试有输出,网络没有问题

# 1. 来用tensorflow进行可视化
with SummaryWriter("./log", comment="sample_model_visualization") as sw:
    sw.add_graph(model, sampledata)

# 2. 保存成pt文件后进行可视化
torch.save(model, "./log/regnet.pth")
torch.save(model.state_dict(), "./log/regnet.pt")
# 3. 使用graphviz进行可视化
out = model(sampledata)
g = make_dot(out)
g.render('modelviz', view=False)  # 这种方式会生成一个pdf文件

(1)可以利用render生成网络图

?(2)可以利用tensorboard生成网络结构图

?

?

  人工智能 最新文章
ros-melodic安装解决sudo rosdep init问题
ner数据扩充脚本例子
机器学习sklearn-逻辑回归制作评分卡
梯度下降代码
常见激活函数及tensorflow使用
torch.squeeze()和torch.unsqueeze()简要介
GLUE标准数据集
机器学习(九)------K-Means实现鸢尾花聚类
Miniconda安装pytorch
opencv图片处理
上一篇文章      下一篇文章      查看所有文章
加:2021-10-01 16:51:20  更:2021-10-01 16:51:38 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
360图书馆 购物 三丰科技 阅读网 日历 万年历 2022年1日历 -2022/1/20 0:42:05-
图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码