随笔分类 - 深度学习
摘要：IOU Jaccard系数（Jaccard index）可以衡量两个集合的相似度。给定集合和，它们的Jaccard系数即二者交集大小除以二者并集大小： $$J(\mathcal{A},\mathcal{B}) = \frac{\left|\math
阅读全文
摘要：参照官方教程，实现pytorch自定义算子。主要分为以下几步： 改写算子为torch C++版本 注册算子 编译算子生成库文件 调用自定义算子 一、改写算子 这里参照官网例子，结合openCV实现仿射变换，C++代码如下： 点击展开warpPerspective.cpp #include "torc
阅读全文
摘要：以下都是pytorch模型转换为onnx，然后利用opencv中dnn模块的readNetFromONNX()函数调用后出现的一些问题， 利用onnxruntime库已经验证onnx结果正确。 相关环境 python: 3.7.4 torch: 1.5.0 onnx: 1.10.2 onnxrunt
阅读全文
摘要：一、Type Error: Type 'tensor(bool)' of input parameter (121) of operator (ScatterND) in node (ScatterND_128) is invalid 问题 模型转出成功后，用onnxruntime加载，出现不支持参
阅读全文
摘要：一、Linux CMakeLists链接版本 官网下载Linux版本libtorch的时候会发现有(Pre-cxx11 ABI)和(cxx11 ABI)两个版本。 如果链接(cxx11 ABI)版本需要在CMakeLists.txt中加入 add_definitions(-D _GLIBCXX_US
阅读全文
摘要：一、All inputs of range must be ints, found Tensor in argument 0: 问题 参数类型不正确，函数的默认参数是tensor 解决措施 函数传入参数不是tensor需要注明类型 我的问题是传入参数npoint是一个int类型，没有注明会报错，更改
阅读全文
摘要：One-hot编码 将标签转换为one-hot编码形式 def to_categorical(y, num_classes): """ 1-hot encodes a tensor """ new_y = torch.eye(num_classes)[y.cpu().data.numpy(), ]
阅读全文
摘要：img_path = "./data/img_37.jpg" # transforms.ToTensor() transform1 = transforms.Compose([ transforms.ToTensor(), # range [0, 255] -> [0.0,1.0] ] ) ## o
阅读全文
摘要：使用OpenCV的DNN模块调用pytorch训练的分类模型，这里记录一下中间的流程，主要分为模型训练，模型转换和OpenCV调用三步。 一、训练二分类模型 准备二分类数据，直接使用torchvision.models中的resnet18网络，主要编写的地方是自定义数据类中的__getitem__，
阅读全文
摘要：一、Ubuntu 系统 系统安装参考链接： https://www.cnblogs.com/masbay/p/11627727.html Dell G3 3590，在自带win10上安装Ubuntu18.04时出现找不到硬盘分区的问题： 解决办法：更改硬盘模式(默认为RAID) 首先进入原有wind
阅读全文
摘要：利用pyinstaller(4.2)打包pytorch，开始使用的python版本为3.7.4，在Ubuntu18.04上能打包成功，但在windows10上一直报错numpy.core.multiarray failed to import，尝试了很多方法，最终在import torch之前添加i
阅读全文
摘要：参考链接 如何计算模型以及中间变量的显存占用大小： https://oldpan.me/archives/how-to-calculate-gpu-memory 如何在Pytorch中精细化利用显存： https://oldpan.me/archives/how-to-use-memory-pyto
阅读全文
摘要：在Mask Rcnn中传给模型训练的是物体的mask图，但训练中会利用到包含物体的平行框，查看代码后发现gt_box是由mask生成的，其主要思路是找出平行框左上角和右下角的坐标。mask图是一个二值图，利用numpy库的where()和any()函数就可以找出和以及和
阅读全文
摘要：目录 一、残差块(Residual Block) 二、 残差网络为什么有用 三、ResNet网络结构 四、代码实现 ——————————————————————————————————————————————————————————————————————————————— 论文：https://a
阅读全文
摘要：一、问题： 最近在keras下利用两个GPU训练Mask_Rcnn模型的时候出现以下错误。 error:RuntimeError: It looks like you are subclassing 'Model' and you forgot to call 'super(YourClass, s
阅读全文
摘要：一、问题： 最近在利用keras训练模型的时候出现以下错误，开始一看以为是cuDNN的版本不对，更换后还是没有效果，后面在网上查找后发现是显存的问题，更改为动态分配内存就可以了。 二、解决方法： 在训练的脚本开头添加以下代码进行动态分配内存，然后重新运行就可以。 import tensorflow
阅读全文
摘要：最近在利用SSD检测物体时，由于实际项目要求，需要对模型进行轻量化，所以考虑利用轻量网络替换原本的骨架VGG16，查找一些资料后最终采用了google开源的mobileNetV2。这里对学习mobileNet系列的过程做一些总结。mobileNetV1是由google在2017年发布的一个轻量级深度
阅读全文
摘要：原文链接：https://blog.csdn.net/weixin_37251044/article/details/81157344 1.什么是均值？ 对于每帧图像来说，均值分为两种：image mean 和 pixel mean。 image mean： 简单的说，读入一张彩色图像，假设是(NN
阅读全文
摘要：最近在利用SSD检测物体时，由于实际项目要求，需要对模型进行轻量化，所以考虑利用轻量网络替换原本的骨架VGG16，查找一些资料后最终采用了google开源的mobileNetV2。这里对学习mobileNet系列的过程做一些总结。mobileNetV1是由google在2017年发布的一个轻量级深度
阅读全文
