栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

pytorch

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

pytorch

TensorBoard使用(我理解是图像显示)
from torch.utils.tensorboard import SummaryWriter
from PIL import Image
import numpy as np
writer=SummaryWriter("logs")  # 创建日志文件夹
writer.add_image("test",img_array,2,dataformats='HWC')   # table ,input,索引

dataformats在数据类型转化的时候使用

for i in range(100): # 自己绘图,y=2*x
    writer.add_scalar("y=2x",2*i,i)  # title y轴  x轴
writer.close() # 关闭

运行之后,进入pycharm的terminal

输入 tensorboard --logdir=logs  进入主机名对应的窗口

若主机服务器多人使用,打开同一个端口了,可以选择指定一个端口

tensorboard --logdir=logs --port=6007  打开6007的端口了

torchvision中的transforms(工具箱,是torchvision用来处理图像的)
from torchvision import transforms

工具线中的很多工具

创建具体的工具  tool=transforms.ToTensor()

使用工具             result=tool(input)

下载数据集时,遇到数据集下载速度慢的,可以将下载链接复制到迅雷下载

  • 卷积层
  • 池化层
  • 非线性激活

下载的训练数据集保存地址

Downloading: "https://download.pytorch.org/models/vgg16-397923af.pth" to C:Usersuser/.cachetorchhubcheckpointsvgg16-397923af.pth
 

调用cuda时用  if torch.cuda.is_available():进行判断是否能够调用gpu进行加速

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/445412.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号