自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(49)
  • 收藏
  • 关注

原创 ubantu16.04 Tensorrt-onnx 安装记录

cmake Could NOT find Protobuf (missing: Protobuf_LIBRARIES Protobuf_INCLUDE_DIR)sudo apt-get install protobuf-compiler libprotobuf-devShapedWeights.hpp:25:21: fatal error: NvInfer.h: No such file or directoryinclude_directories(/home/wzp/project/trt/Te

2021-04-23 18:28:51 1676

原创 Expected object of scalar type Double but got scalar type Float for argument

问题描述:pytorchx=np.zeros((1,3,512,512))x=torch.from_numpy(x)out=net(x)错误提示:Expected object of scalar type Double but got scalar type Float for argument解决方案:添加x=x.float()

2021-01-19 20:36:32 678

原创 关于Pytorch的N个使用心得

前言这是一篇pytorch使用心得的文章,是一个筑基期AI灵魂工程师的手册!当然对于大能来说,都是基础的不能再基础的东西,修行之路漫漫呀!1.网络训练函数也可以建立为1个类以前我觉得训练模型的话,写一个训练函数就行了,后来看到别人把训练函数写成一个类。 为什么要写成一个类? 答:因为可以封装其他操作,使用比较方便,调用也比较方便2.Aloss+Bloss=Closs,反向传播时,只需要传Closs目前接触的大部分为单损失函数,所以当有两个函数时,我在纠结怎么写反向传播呢?是分别传递

2021-01-09 11:46:50 379

原创 ‘numpy.ndanrray‘ object has no attribute ‘permute‘

问题描述:对numpy.ndarray格式数据进行维度转换,报错:'numpy.ndanrray' object has no attribute 'permute'原因分析:如提示所言,numpy.ndanrray没有permute这个属性解决方案:转换为Tensor,Tensor具有这个属性tensordata = torch.from_numpy(numpydata)...

2020-11-27 10:04:05 13672 5

原创 Pytorch关于池化步长为元组的探究

看了很多关于池化的文章,发现都没有谈到池化步长为元组的情况,索性自己做实验探究一番最常见的池化,尺寸为2,步长为2,很容易理解,即把(w,h)变为(w/2,h/2),那如果是池化的尺寸为(1, 2), 步长为(2, 1)呢?怎么理解?首先尺寸(1,2)应该是这样的形状: 步长(2,1)这应该是两个方向的步长,纵向...

2020-10-13 10:50:14 1286

转载 基于python自动下载百度图片

#-*- coding:utf-8 -*-import reimport os#import requestsimport urllib.requestimport socketsocket.setdefaulttimeout(5)keyword = ""i = 0def dowmloadPic(html, word): global i pic_url = re.findall('"objURL":"(.*?)",',html,re.S) print ...

2020-10-11 16:43:06 187

原创 CV函数错误:error: (-215:Assertion failed) src.type() == (((0) & ((1 << 3) - 1)) + (((1)

问题描述:在使用cv2.threshold这个函数时弹出错误error: (-215:Assertion failed) src.type() == (((0) & ((1 << 3) - 1)) + (((1)原因分析:cv2.threshold函数要求输入图像为灰度图,我之前也确实做了灰度转换img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) img = img.astype('float32')但是我又将灰度...

2020-10-09 09:46:18 3392

原创 探究pytorch model.eval()测试效果远差于model.train()

探究pytorch model.eval()测试效果远差于model.train()

2020-08-20 11:09:05 5063 8

转载 tensor、numpy.array、lis的互相转换

1.1 list 转 numpyndarray = np.array(list)1.2 numpy 转 listlist = ndarray.tolist()2.1 list 转 torch.Tensortensor=torch.Tensor(list)2.2 torch.Tensor 转 list先转numpy,后转listlist = tensor.numpy().tolist()3.1 torch.Tensor 转 numpyndarray = tenso...

2020-08-17 18:00:39 362

空空如也

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除