Pytorch学习
(^_^)Alex
这个作者很懒,什么都没留下…
展开
-
Torchvision数据集下载与使用
1.单张图片处理import torchvisiontrain_set = torchvision.datasets.CIFAR10(root="./dataset", train=True, download=False)test_set = torchvision.datasets.CIFAR10(root="./dataset", train=False, download=False)img, label = test_set[4] # img是图片的信息, label是标签信息(用整数表原创 2022-04-16 15:16:10 · 1310 阅读 · 0 评论 -
常见Transform操作
from PIL import Imagefrom torch.utils.tensorboard import SummaryWriterfrom torchvision import transformswriter = SummaryWriter("logs")img_path = "dataset/train/ants/9715481_b3cb4114ff.jpg" # 图片路径img_PIL = Image.open(img_path)原创 2022-04-16 10:56:16 · 351 阅读 · 0 评论 -
4.Transform的使用
from torch.utils.tensorboard import SummaryWriterwriter = SummaryWriter("logs")for i in range(50): writer.add_scalar("y=i^2", i*i, i)writer.close()from PIL import Imagefrom torch.utils.tensorboard import SummaryWriterfrom torchvision i...原创 2022-04-16 09:58:37 · 89 阅读 · 0 评论 -
3.TensorBoard
1.PyCharm输入from torch.utils.tensorboard import SummaryWriterwriter = SummaryWriter("logs")for i in range(50): writer.add_scalar("y=i^2", i*i, i)writer.close()2.浏览器中显示,在控制台中输入–logdir指定目录,–port指定端口tensorboard --logdir=logs --port=6009先运行pych原创 2022-04-16 08:52:49 · 90 阅读 · 0 评论 -
2.加载数据集
1.下载数据集,放在项目的目录下项目名右键–>open in explorer–>将数据集拷贝到该目录下2.(1)from PIL import Image //导入包(2)设置图片路径方法1:windows双斜杠进行转义img_path="D:\\Code\\Python\\learnPytorch\\dataset\\train\\ants\\0013035.jpg"方法2:引号前加r避免转义img_path=r"D:\Code\Python\learnPytorch\d原创 2022-04-15 22:21:53 · 917 阅读 · 0 评论 -
1.Pytorch环境配置
1.下载Anaconda2.打开Anaconda Prompt,显示(base)表明安装成功.3.管理环境,Prompt中输入conda create -n pytorch python=3.8.8,接着输入y安装必要的包4.激活pytorch环境,输入 conda activate pytorch5.显示已安装的包,输入 pip list6.下载pytorch,Start Locally | PyTorch选择稳定版,安装命令 conda install pytorch t...原创 2022-04-15 17:51:50 · 618 阅读 · 0 评论