- 博客(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关注的人
RSS订阅