西瓜撞树上了-CSDN博客

原创 OrienMask TensorRt

文章目录Pre ProcessCreate EngineInferencePost Process总结在之前一篇博文：《Real-time Instance Segmentation with Discriminative Orientation Maps》中介绍了OrienMask的Python代码，在这里补上Tensorrt下推理OrienMask。Pre Process在OrienMask里预处理阶段比较简单，直接做了resize到模型输入尺寸再除以255做归一化，做完图像转换，将结果存到CUD

2022-02-11 19:12:48 3082 3

原创 Real-time Instance Segmentation with Discriminative Orientation Maps

目录网络结构网络输出后处理总结OrienMask是2021年新出的一篇实时的实例分割模型。下图是OrienMask在coco数据集上的测试结果。在单张2080ti上可以达到42.7fps的速度。这篇文章的主体思路是先使用yolo检测出目标的bbox，然后再检测出bbox中的前景来完成实例分割任务。网络结构上图是OrienMask的网络结构，主要分为3块：Backbone、FPN、Heads。使用darknet53作为Backbone，进行了32倍下采样，分别在4x，8x，16x，32x下采样的

2021-11-03 17:29:28 911

原创 yolov5 自适应padding

yolo系列输入尺寸大多是416*416，608*608这种正方形尺寸的。图像预处理中通常先把图像padding成正方形然后在reshape到指定尺寸，但是这样会在图像中产生很大一部分无效区域。在yolov5中提出了一种自适应padding方法，在预处理过程中对原始图像自适应地填充最少的黑边。先看源码def letterbox(im, new_shape=(640, 640), color=(114, 114, 114), auto=True, scaleFill=False, scaleup=True

2021-11-01 17:08:46 2898

原创 TensorRT safety runtime

在tensorrt上移植深度学习模型的时候需要在其安全模式下进行移植。有好多坑，这里记录一下tensorrt安全模式下的一些限制。：TensorRT safety runtime 不支持DLAsafety runtime 不支持implicit batch size。batch size 必须是explicit。创建configurations时必须包含NetworkDefinitionCreationFlag::kEXPLICIT_BATCH .safety runtime不支持动态尺寸safe

2021-10-27 22:27:58 560

原创 ubuntu18.04下安装docker、NVIDIA-docker

文章目录docker安装NVIDIA-docker安装免sudo使用docker卸载dockerdocker常用命令docker是一个开源的应用程序引擎。在开发过程中我们经常遇到复杂的环境配置，docker可以帮助我们解决这个问题：可以现在在docker容器中搭建环境，然后将这个容器提交成镜像，然后镜像可以保存成tar等格式的压缩文件；当我们更换机器的时候我们只需要加载这个存着镜像的压缩文件就可以创建相同的镜像，再根据镜像穿件一个容器就可以完美复制之前的环境。下面来介绍下怎么在ubuntu18.04上安

2021-10-21 23:08:35 822

原创 Git常用操作

Git常用操作git branch <branch name> #创建分支git checkout <branch name> #切换到指定分支git pull origin <branch name> # 更新代码# 提交代码三部曲git status # 查看自己改动了哪些东西。git add . #向缓冲区添加所有修改内容git commit -m "beizhu" # 提交并记录备注git push origin <branch nam

2021-08-19 19:33:31 80

原创 docker常用操作【持续更新】

docker常用操作保存镜像：# step1: commit containerdocker commit -a "xiaobo" -m "ground_aware_env" 00e037a9d444 tensorrt7.1:ground_aware_plugin #commit containerdocker commit --help -a, --author string Author (e.g., "John Hannibal Smith <hannibal@a-team

2021-08-19 19:32:21 113

原创 2021-08-19 Ubuntu下安装opencv

opencv 安装下载opencv4.2.0 , 并放在/home目录下更新软件源并安装依赖sudo apt-get update sudo apt-get install build-essential cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev 编译sudo unzip opencv-4.2.0.zipcd opencv-4.2.0mkdir buil

2021-08-19 19:28:49 132

原创 pytorch多卡训练显卡利用率问题

多卡训练显卡利用率问题最近我继承了前同事的深度学习代码，使用pytorch写的。在nvidia-smi查看显卡利用率的时候发现显卡是经常出现除了第一张显卡外，其他7张显卡的利用率为0的情况。同时查看了CPU利用率，发现大多的核也是空闲的：阅读代码后先后试了：调整dataloader的num_workers的数量（之前为默认值）、把数据先加载到内存的方法均无法得到明显改善。然后我debug各个阶段的耗时，发现在inference时耗时0.1秒多，但是在计算loss和回传loss时总共用时接近6秒（

2021-04-19 11:52:43 3597 2

原创 libtorch at::Tensor转int

libtorch at::Tensor转inttorch::Tensor tmp = torch::ones({3,3});int t = tmp[0][0].item<int>();

2021-03-10 17:07:46 3059

原创 mobile V1到V3

目录MobileNet-V1概要深度可分离卷积深度可分离卷积pytorch代码MobileNet-V1论文地址概要作者将传统卷积分成了两步：depth wise convolution和point wise convolution。字面理解是先深度分离地逐层卷积，然后使用逐点的1*1卷积将深度分离的特征连接起来。这样大大降低了计算复杂度。引入了width multiplier和resolution multiplier两个超参数。width multiplier（宽度系数）是用来减少输入的特征通

2020-11-19 17:04:31 661

原创 pytorch计算模型算力与参数大小

目录ptflops介绍get_model_complexity_info()栗子ptflops介绍官方链接这个脚本设计用于计算卷积神经网络中乘法-加法操作的理论数量。它还可以计算参数的数量和打印给定网络的每层计算成本。支持layer：Conv1d/2d/3d，ConvTranspose2d，BatchNorm1d/2d/3d，激活(ReLU, PReLU, ELU, ReLU6, LeakyReLU)，Linear，Upsample，Poolings (AvgPool1d/2d/3d、MaxPool

2020-11-17 17:09:32 8124 2

weixin_47062350的博客