![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
yuejich
温故而知新,可以为师矣!!
展开
-
RPN的理解,大家来指正
vgg16-input:一张图片,8008003的rgb图像特征提取层:vgg16,16倍下采样vgg16-output:5050512的feature maprpn-input=vgg16-output=5050512rpn的第一层可以理解为256个33的卷积核对feature map进行滑窗rpn-output1=5050256或者说是2565050,anchor的生成,可以认为是在5050256的特征图上的每一个像素点生成3个尺度3个比例的anchor box,也就是9个anchor bo原创 2021-08-12 19:57:05 · 238 阅读 · 0 评论 -
学习总结:关于pytorch保存save和加载load模型的总结
保存权重,推荐使用这种方法,不容易出错 torch.save(model.state_dict(), 'wbdog.pt') print('Finish training')上面的这个代码用了torch.save()方法,这个只是保存了训练好的权重到wbdog.pt文件保存模型以及权重 torch.save(model, 'wbdog.pt') print('Finish training')这样就把模型也保存了,...原创 2021-08-02 18:35:48 · 441 阅读 · 0 评论 -
深度学习实战——基于pytorch写的一个猫狗分类的模型——学习总结
写在前面本学习总结主要目的1.怕以后忘记,到时候翻来看看,2.梳理一下项目过程,便于工业化生产(哈哈)3.敲一遍代码加深记忆4.帮助他人,方便入门同学能够容易入门本过程主要设计到的内容1.torch.nn,nn模块下的Module类,组件类,neture network2.tensor,巩固tensor的方法3.torch.utils.data里面DataLoader的用法4.torchvision里面transforms的用法5.torchvision.datasets里面Image原创 2021-07-23 17:07:41 · 2673 阅读 · 15 评论 -
学习笔记——pytorch怎么把PIL读取的图像转换成tensor,并归一化处理
torchvision的transforms的使用总结import torchfrom torchvision import transformsfrom PIL import Imageimage=Image.open('dog.jpg')print(image.size)toTensor=transforms.ToTensor()# 实例化一个toTensorimage_tensor=toTensor(image)print(image_tensor.size())...原创 2021-07-23 14:52:26 · 2174 阅读 · 0 评论 -
学习笔记——pytorch的torchvision.datasets的ImageFolder使用
from torchvision.datasets import ImageFolderimport matplotlib.pyplot as pltimport torchvision.transforms as T#总结一下torchvision.datasets的ImageFolder类root="catanddogs_dataset"#ImageFolder是一个class,该类的初始化方法需要传入5个参数,第一个参数root是一个string类型的,需要传入图片文件夹的path'''原创 2021-07-23 11:55:25 · 1059 阅读 · 1 评论