深度学习
关注
分享
扫一扫
文章平均质量分 63
幸福回头
A Sunny Boy (BJTU)
展开
-
PyTorch和ONNX模型相互转换
模型一共有三层，两层卷积运算，一层全连接，每个卷积运算的卷积核和输入尺寸和输出尺寸都可以看到。但是看到这层网络有一个问题，在全连接之前，缺少一个Reshape算子，将1*14*8*8的特征图变成一维向量，将向量送入全连接，得到相应的输出【1,10】目前遇到一个问题，有一个ONNX的网络模型，可以通过Netron进行可视化，Netron地址为。1. 根据可视化的网络模型进行Pytorch框架的网络模型重建（结构）使用Parameter进行权重的赋值。最后的函数是将PyTorch模型转换为ONNX。原创 2023-06-14 16:10:43 · 2420 阅读 · 0 评论 -
ONNX&QLinearConv量化卷积详解
主要讲解ONNX以及其中的线性量化卷积运算流程原创 2022-08-12 11:16:45 · 2075 阅读 · 0 评论 -
mmsegmentation 报错
mmsegmentation报错原创 2022-04-20 18:05:05 · 3321 阅读 · 1 评论 -
mmdetection3d 踩坑记录
mmdetection3d的踩坑记录原创 2022-03-29 18:41:56 · 3018 阅读 · 1 评论 -
自动驾驶车载激光雷达-数据集整理
搜索了好久，发现目前网上没有统计好的车端自动驾驶数据集的信息，根据目前我搜索到 的一些车端自动驾驶激光雷达数据集进行整理，形成如下的表格，欢迎大家补充信息。自动驾驶车端激光雷达数据集统计 数据集名称 出品方 激光雷达线数 数量 图像 KITTI 德国卡尔斯鲁厄理工学院&丰田美国技术研究院 64 8k √ LiDAR-Video Driving Dataset 上海交通大学&a...原创 2021-12-14 16:21:42 · 4369 阅读 · 0 评论 -
conda虚拟环境路径包含系统python路径问题
趁着618阿里云搞活动，买了一个阿里云最低配版本的ECS服务器，想着给一个内部的服务器做一个内网穿透，这样可以在任何地方连接到服务器进行操作。、原创 2021-08-04 09:15:39 · 3458 阅读 · 0 评论 -
KITTI数据集 激光雷达-图像坐标系转换关系
关于KITTI坐标系中的坐标转换，研究了好久，网络上也没有很详细的解释，自己了解了一些转换的内容，写在这里，供大家参考学习。KITTI数据集中一共有三个坐标系：1. 激光雷达坐标系 (下图1中的蓝色坐标系)2. 相机坐标系 (下图1中的红色坐标系)3. 图像坐标系 (下图2相机采集的图像) ...原创 2021-03-15 17:28:03 · 11841 阅读 · 23 评论 -
mmdetection3D报错AttributeError: module ‘pycocotools‘ has no attribute ‘__version__‘
在安装完mmdetection3D之后，使用其中现有数据集的训练脚本进行训练时候，出现了以下错误： AttributeError: module 'pycocotools' has no attribute '__version__' 其实是因为在安装mmdetection的时候，其中系统上已经安装了pycocotools的python包，而mmdetection是使用的自己开发的pycocotools，当系统中已经安装了此包的时候，mmdetection不会覆盖安装，也就是...原创 2021-02-18 10:15:39 · 4750 阅读 · 8 评论 -
mmdetection3D RuntimeError: Error compiling objects for extension
最近在配置open-mmlab的mmdetection3d项目，其中大部分的步骤都可以很方便的完成，但是在CUDA版本和torch版本上会有一定的问题，详细记录下： 前面的几个步骤都没什么问题，就最后一步的python setup develop的时候，出现了 RuntimeError: Error compiling objects for extension 这个错误 https://github.com/sshaoshuai/Pointnet2.P...原创 2021-02-03 16:25:58 · 28573 阅读 · 15 评论 -
Numba Ubuntu包安装指南(含llvm、llvmlite，最简单)
最近在研究点云和图像融合的深度学习算法，目前正在配置相关的算法上的代码，numba这个包需要依赖llvm以及llvmlite，而这两个包安装起来很麻烦，现有网络上没有详细的相关指南，能查到的指南都是17年的安装指南了，版本太多老旧，导致numba的包不兼容，现将正确的安装方法进行记录，希望也能帮助其他人少走一些坑。 想要安装numba必须要安装llvm，具体的安装方法如下:$ wget -c http://releases.llvm.org/7.0.1/clang+l...原创 2020-08-25 15:47:00 · 9672 阅读 · 13 评论 -
failed to run cuBLAS routine cublasSgemm_v2 CUDA报错
在配置服务的过程中，安装好了CUDA9.0和cuDNN之后，想着把代码上传到服务器上进行训练一波测试下RTX 2080TI的速度，谁想到，竟然报了个错误，错误如下： failed to run cuBLAS routine cublasSgemm_v2 环境cuda9.0+tensorflow1.12.0 Ubuntu16.04 LTS 显卡 RTX 2080Ti 找了下原因： NVIDIA cuda 9.0自身的bug 解决方法安装cu...原创 2020-05-15 10:49:47 · 2353 阅读 · 2 评论 -
服务器Deep Learning环境Inference搭建(Python3、Tensorflow1.12、Keras、opencv)
搭建CentOS7.4DL环境，服务器是为VCAA卡驱动的(通过PCIe插到服务器上)，VCAA主要目的是利用VPU实现Inference推理加速。首先为了验证模型可用，就在服务器上搭建了普通的CPU环境以检验模型可用性。 1.查询Python版本 首先是服务器的CentOS中是含有Python的，但是可以查看一下它的版本是python2的，并不是pytho...原创 2020-02-13 12:05:29 · 667 阅读 · 0 评论 -
Jetson——TX2 开机设置相关问题(工程问题)开机启动脚本，默认启动风扇
因为实验室要将TX2做成相关的产品，要将TX2修改成上电自启动并且运行相关程序，踩了一些坑，相关的记录发表在CSDN上，也有一段时间没写博客了（忙于找工作和各种面试），终于马上要卖身成功，补上最近的博文空白。 TX2上电自启 TX2开机之后的风扇是需要手动输入指令来启动的，要将设备做成全自动的，也就是上电自动启动的界面，在网上查了好多，也做了很多不同的测试，可...原创 2019-12-16 18:27:58 · 4476 阅读 · 1 评论 -
TensorRT6.0.1安装指南
看到2019年9月18号，TensorRT6.0.1发布了，也是偶然看到了，发现这个版本已经从TensorRT5--->TensorRT6，那肯定是有重大性能上的提升，安装的方法以及出现的问题在这列举一下，以防止后人继续踩坑。 首先要下载TensorRT6.0.1版本，直接去官网下载即可，需要有一个NVIDIA 的官方账号，登录之后填写一个相关内容就可以进行...原创 2019-09-18 11:03:57 · 8989 阅读 · 8 评论 -
配置Tusimple的MXNet踩坑记
首先就是从Tusimple的Github开源项目中的INSTALL.md进行，发现里面有Setup with Docker$HOST-SIMPLEDET-DIR:$CONTAINER-WORKDIR registry.cn-beijing.aliyuncs.com/rogerchen/simpledet:cuda9 zsh 首先要安装nvidia-dockerwget -P /...原创 2019-08-28 15:55:30 · 398 阅读 · 0 评论 -
Mask_RCNN的TensorRT加速
现有的网络上，主流的Mask RCNN主要有两个，一个是matterport的Keras版本https://github.com/matterport/Mask_RCNN，有13.5K的star，一个是facebookresearch的PyTorch版本6.2K的star，链接如下https://github.com/facebookresearch/maskrcnn-bench...原创 2019-08-14 13:50:59 · 6892 阅读 · 24 评论 -
Onnx-TensorRT安装指南(踩坑记)
首先要来到onnx-tensorrt的Github官网Github链接 下面就是步骤git clone --recursive https://github.com/onnx/onnx-tensorrt.gitmkdir buildcd buildcmake .. -DTENSORRT_ROOT=<tensorrt_install_dir&g...原创 2019-08-07 16:28:25 · 12175 阅读 · 2 评论 -
Pytorch转Onnx转TensorRT踩坑记
转换Onnx过程中：PyTorch v1.0.1 Reshape不支持报错源码安装PyTorch v1.0.1 caffe2报错PyTorch v1.2.0 需要升级cuda10.0以上PyTorch v1.1.0报错：RuntimeError: Failed to export an ONNX attribute, since it's not constant, ple...原创 2019-08-13 17:25:20 · 7882 阅读 · 0 评论 -
火车轨道铁路轨道检测识别(附带Python源码+详细解析)
现在的网络上，铁轨检测的源码几乎没有，所以自己参照着一篇汽车车道线检测的方法，然后调节参数，实现了铁轨的轨道检测，但现在只能检测直线，弯曲的铁轨检测下一步会实现，实现之后会更新的，敬请期待。 弯轨检测的已经实现并且检测效果不弱于直线：火车轨道铁路轨道检测识别（弯轨+直轨）通用性（Python源码+讲解） 针对现有的火车轨道检测，从汽车车道线延伸过来，用的是同样的...原创 2019-03-15 09:47:53 · 21231 阅读 · 20 评论 -
TX2 （JetPack4.2）安装 Pycharm&TensorFlow
踩了两天的天坑，总算是把这个大神软件装上了，网络上各种不同的说法，我几乎都尝试了一遍，有些版本不一样的，安装方式都不相同，这里先说明我的安装版本： JetPack4.2 刚开始的时候，天真的以为，像ubuntu一样，我可以直接下载的软件解压之后，就可以顺利的安装，后来经历的天坑之后，才发现自己太年轻。 首先，将下载的tar文件解压，找到bin文件夹，利用s...原创 2019-03-29 20:28:56 · 7739 阅读 · 17 评论 -
Jetson TX2 下安装pillow 报ImportError: The _imagingft C module is not installed方案
Jetson TX2上的Ubuntu18.04 系统，使用pip安装pillow,运行时出现ImportError错误“The _imagingft C module is not installed”具体出错的那行代码是font = ImageFont.truetype()解决过程如下：先确保代码中导入语句是：from PIL import ImageFont尝试先卸载pil...原创 2019-03-30 11:32:24 · 2498 阅读 · 3 评论 -
Jetson TX2 虚拟内存
一般在TX2上运行的程序，8G的内存（其实就3.5G左右）很容易就被killed，这个时候，就需要虚拟内存来进行扩展swap 1. 创建8G大小的swapfilefallocate -l 8G swapfile2. 更改swapfile的权限chmod 600 swapfile3. 创建swap区mkswap swapfile4. 激活swap区...原创 2019-03-30 12:08:50 · 2452 阅读 · 0 评论 -
Jetson TX2 刷机教程(JetPack4.2版本)
自从NVIDIA出现JetPack4.2(Ubuntu18.04)版本之后，安装方式和之前就大不相同，看了前面的几个安装版本之后，感觉新版的好像安装起来更加简洁了，只需要一个SDK就可以，下面来看看详细步骤： ①下载SDK Manager，从英伟达官网下载 请点击下载地址进行文件下载 下载完成之后是一个.deb文件，安装在......原创 2019-03-27 17:34:38 · 42495 阅读 · 113 评论 -
Jetson TX2 高功率模式更改 Can't access Fan!(JetPack4.2)
在JetPack4.2之前的版本上，在/home文件下会有一个 jetson_clocks.sh，如果想要开启高功率模式，直接使用命令行sudo ~/jetson_clocks.sh 就可以开启风扇，但是在最新的JetPack4.2版本中，这个文件消失了，不得已我从CSDN上下载了一个jetson_clocks.sh，但是，当我使用命令行开启的时候，出现了Ca...原创 2019-03-28 15:52:25 · 5505 阅读 · 1 评论 -
Jetson TX2 安装Keras 出现Failed cleaning build dir for scipy
在ARM架构下的scipy本身就是一个很难安装的软件包 利用pip3 install keras的时候，出现了run scipy setup.py \. 然后蹦出来了Failed cleaning build dir for scipy 查了好多资料，可以安装下面的依赖包，就可以成功的安装keras了sudo apt-get install python3-sc...原创 2019-04-07 17:37:16 · 7460 阅读 · 5 评论 -
BDD100K数据集（内附国内下载链接_1.5M/S）
BDD100K：一个大规模、多样化的驾驶视频数据集 内部包含有1.8T的视频集合 6.5G的目标检测数据集。包括Bus、Light、Sign、Person、Bike、Truck、Motor、Car、Train、Rider总共约有184万个标定框，不同类型目标的数目统计如图所示。 官方网址为：https://bair.berkeley.edu/bl...原创 2019-05-19 10:18:13 · 12158 阅读 · 24 评论 -
常用数据集链接
CVonline：图像数据库(Google直译的结果，希望对大家有帮助)按主题索引行动数据库 属性识别 自主驾驶 生物/医药 相机校准 脸和眼/虹膜数据库 指纹 一般图像 一般RGBD和深度数据集 一般视频 手，掌握，手动和手势数据库 图像，视频和形状数据库检索 对象数据库 人（静），人体姿势 人员检测和跟踪数据库（另见监控） 遥感...转载 2019-05-19 10:57:16 · 8491 阅读 · 1 评论 -
BDD100k数据集提取Json至txt格式(YOLOv3可用)
由于需要，最近想要训练一下全球最大的自动驾驶数据集BDD100K，包括有100k的训练集和10k的验证集 无奈，现有的数据标注形式为Json数据格式，需要转换成为YOLO格式的数据集。 这个数据集的标注是一个大文件，足足有15个G的标注文件，处理起来速度不是很快，10k的val集还好，100k的train集真的速度好慢好慢………… import js...原创 2019-05-19 20:18:14 · 4347 阅读 · 21 评论 -
Keras API调用模型下载一半断网问题
最近使用Keras调用MobileNetv2时候，在使用过程中断了一次网，当时正在下载预训练权重，然后重新调用程序的时候，会出现Error。关于Keras调用模型加载预训练模型的位置问题。 这个时候，是因为当原有的模型加载过程中，不会重新下载，而是尝试去加载下载一半的模型，但是下载一半的模型，不完整，会出现错误，这个时候需要重新删除原来下载完一般的模型，但是这个模型的位置在哪里呢，...原创 2019-05-22 13:17:28 · 428 阅读 · 0 评论 -
深刻解读MobileNetv1网络结构
本文转载自：https://zhuanlan.zhihu.com/p/31551004引言卷积神经网络（CNN）已经普遍应用在计算机视觉领域，并且已经取得了不错的效果。图1为近几年来CNN在ImageNet竞赛的表现，可以看到为了追求分类准确度，模型深度越来越深，...转载 2019-05-22 13:22:39 · 5658 阅读 · 0 评论 -
MobileNet V2 模型分析+解读
一、论文：https://arxiv.org/abs/18转载自https://blog.csdn.net/kangdi7547/article/details/8143157二、论文理解MobileNet V2 是对 MobileNet V1 的改进，同样是一个轻量级卷积神经网络。1）基础理论--深度可分离卷积（DepthWise操作）标准的卷积过程可以看...转载 2019-05-22 13:26:12 · 3009 阅读 · 0 评论 -
Keras|Tensorflow 计算模型的FLOPs
最近在研究模型的计算量，发现Pytorch有库可以直接计算模型的计算量，所以需要一个一个Keras和Tensorflow可以用的，直接把Model接入到函数中，print一下就可以计算出FLOPs FLOPS：注意全大写，是floating point operations per second的缩写，意指每秒浮点运算次数，理解为计算速度。是一个衡量硬件性能的指标。...原创 2019-07-26 15:55:48 · 13208 阅读 · 21 评论 -
VOC格式数据集制作(YOLOv3 可用)
参考链接：使用labelImg制作自己的数据集（VOC2007格式）用于Faster-RCNN训练VOC格式的数据集由五个文件夹组成将所有的图片放入JPEGImages并重命名下面是重命名的C++源码#include <iostream> #include <io.h> //对系统文件进行操作的头文件#include <string>...原创 2019-03-04 10:09:06 · 1933 阅读 · 3 评论