
深度学习框架
nudt_qxx
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
常见深度学习框架和推理框架的通道顺序
N: batchC: channelH: heightW: widthCaffe 的Blob通道顺序是:NCHW;TensorFlow的tensor通道顺序:默认是NHWC, 也支持NCHW,使用cuDNN会更快;Keras和TensorFlow一样,默认使用NHWC,但可以用data_format='channels_last' #NHWCdata_format='channels_first' #NCHW进行切换Pytorch中tensor的通道顺序:NCHWPaddle中原创 2021-09-14 11:26:06 · 449 阅读 · 0 评论 -
自定义Op编译进Tensorflow并使用Py和C++接口调用
2018.07.23成功把自定义op加入TensorFlow并用C++和Python调用,更新文档:克隆最新版Tensorflow,包括依赖:git clone --recurse-submodules https://github.com/tensorflow/tensorflow.git自定义运算的两个cpp,my_add.cc和zero_out.cc,放置于tensorflo...原创 2018-07-24 07:12:03 · 3514 阅读 · 4 评论 -
DL&ML评测之IoU,ROC,AUC和mAP详解
基本概念:1. TP,FP,TN,FNTP:True Positive,预测为正样本_实际也为正样本的特征数FP:False Positive,预测为正样本_实际为负样本的特征数TN:True Negative,预测为负样本_实际也为负样本的特征数FN:False Negative,预测为负样本_实际为正样本的特征数如maskrcnn-benchmark/datasets/jinnan/jinnan2_round1_train_201903052、修改paths_catalog.py路径为maskrcnn-benchmark/maskrcnn_benchmark/config/pa...转载 2019-03-25 14:18:56 · 2126 阅读 · 1 评论 -
mmdetection 多尺度测试开启
当然在issue:https://github.com/open-mmlab/mmdetection/issues/135 里面陈凯博士说的很清楚,这里以faster-rcnn为例,看下如何开启多尺度测试提分:文件准备git clone https://github.com/open-mmlab/mmdetectioncd ..mkdir models && cd mode...原创 2019-06-01 15:43:37 · 7304 阅读 · 13 评论 -
mmdetection加入SENet-154 backbone
代码搬用自:https://github.com/open-mmlab/mmdetection/issues/5881. 在mmdet/models/backbones中新建senet.py,加入以下内容:from __future__ import print_function, division, absolute_importfrom collections import Ordere...原创 2019-06-02 14:46:16 · 6438 阅读 · 10 评论 -
浅谈 PyTorch 中的 tensor 及使用
原文出处:浅谈 PyTorch 中的 tensor 及使用:https://zhuanlan.zhihu.com/p/67184419这篇文章主要是围绕 PyTorch 中的 tensor 展开的,讨论了张量的求导机制,在不同设备之间的转换,神经网络中权重的更新等内容。面向的读者是使用过 PyTorch 一段时间的用户。本文中的代码例子基于 Python 3 和 PyTorch 1.1,如果文...转载 2019-06-08 16:47:03 · 1275 阅读 · 0 评论 -
Pytorch加速Dataloader提升数据读取速度
出处:https://zhuanlan.zhihu.com/p/66145913在利用DL解决图像问题时,影响训练效率最大的有时候是GPU,有时候也可能是CPU和你的磁盘。很多设计不当的任务,在训练神经网络的时候,大部分时间都是在从磁盘中读取数据,而不是做 Backpropagation 。这种症状的体现是使用 Nividia-smi 查看 GPU 使用率时,Memory-Usage 占用率很...转载 2019-06-08 16:53:21 · 24834 阅读 · 4 评论 -
C++调用TensorFlow和TensorRT进行加速
本文系转载,出处:如何用CPP调用TensorFlow&&TensorRT环境tensorflow r.1.13 + tensorrt-5.0.2.6cuda10cudnn 7.6.1初步接触最近训练了一个人脸的landmark regression&& confidence output的model( based Tensorflow)。因为模型中...转载 2019-07-03 21:08:10 · 8705 阅读 · 11 评论 -
在Tensorflow新增一个自定义Op
本文系转载出处:http://wiki.jikexueyuan.com/project/tensorflow-zh/how_tos/adding_an_op.html预备技能对 C++ 有一定了解.已经下载 TensorFlow 源代码并有能力编译它.如果现有的库没有涵盖你想要的操作, 你可以自己定制一个. 为了使定制的 Op 能够兼容原有的库 , 你必须做以下工作:在...转载 2018-07-11 16:03:39 · 10013 阅读 · 2 评论 -
Pytorch实现手写数字mnist识别
import torchimport torchvision as tvimport torchvision.transforms as transformsimport torch.nn as nnimport torch.optim as optimimport argparse# 定义是否使用GPUdevice = torch.device("cuda" if torch.cu...原创 2018-06-17 17:08:56 · 2560 阅读 · 1 评论 -
gluon/mxnet实现mnist手写数字识别
import gluonbook as gbfrom mxnet import autograd,nd,init,gluonfrom mxnet.gluon import loss as gloss,data as gdata,nn,utils as gutilsimport mxnet as mxnet = nn.Sequential()with net.name_scope():...原创 2018-06-17 16:53:26 · 1195 阅读 · 0 评论 -
win10下导入theano报错
解决方案为: 先卸载:pip uninstall theano 然后安装mingw和libpython:conda install mingw libpython 接下来import theano就不会报错了!原创 2017-05-28 09:52:05 · 987 阅读 · 0 评论 -
win10 python3.5.2下安装facenet
Google研究人员在2015年发布了一篇关于人脸识别的论文:FaceNet: A Unified Embedding for Face Recognition and Clustering,然后OpenFace是首个应用谷歌FaceNet深度神经网络进行面部识别的开源项目,地址openface。 后来又有大神使用tensorflow重写了openface,该项目是facenet,地址facene原创 2017-05-30 13:13:20 · 7832 阅读 · 5 评论 -
Ubuntu16.04 Python 3.6安装Pytorch 0.2
这里选择用pip方式安装:wget http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whlpip install torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl然后即可安装成功!原创 2017-11-29 10:21:45 · 3557 阅读 · 1 评论 -
Windows下安装PyTorch0.3.0
更新提醒:本文已过期,PyTorch0.4.0已经有官方的Windows支持,Windows下安装最新的PyTorch0.4.0请移步本人另一篇博客:Windows下安装PyTorch0.4.0。本文系转载,出处:关于 PyTorch 0.3.0 在Windows下的安装和使用。PyTorch简介在2017年1月18日,facebook下的Torch7团队宣布PyTorch开源后就引来...转载 2017-12-06 23:21:10 · 20393 阅读 · 11 评论 -
编译Mxnet出错/usr/bin/ccache: invalid option -- 'E'
环境Ubuntu 14.04 + Cmake3.2.2今天用Cmake编译MxNet,报错/usr/bin/ccache: invalid option -- 'E',全部日志如下:[ 0%] Building NVCC (Device) object CMakeFiles/cuda_compile.dir/src/operator/cuda_compile_generated_pad.cu.o原创 2018-01-03 10:57:12 · 1417 阅读 · 0 评论 -
实时语义分割框架FCIS的官方代码安装和运行
介绍继图像分类、物体检测之后,精确到像素级别的物体实例分割就成为更具挑战性和实用性的视觉识别任务。前两个任务在近年来取得了迅速的进展,已经有了不少优雅有效的方法。然而,实例分割任务却还缺少一个标杆性的工作。为此,来自清华大学和微软研究院的李益、齐浩之、代季峰、季向阳、危夷晨合作,利用全卷积神经网络(FCN)在图像语义分割和实例分割预测方面的优势,提出了一种新的架构 FCIS。FCIS 是首个全卷积、原创 2018-01-04 15:23:50 · 4692 阅读 · 4 评论 -
Ubuntu16.04下cuda8.0.run和cuda9.0文件下载
wget https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda_8.0.61_375.26_linux-run原创 2018-01-13 23:26:47 · 3075 阅读 · 0 评论 -
Windows下安装MXNet的动态图接口Gluon
新闻2017年10月12日,微软和亚马逊又达成一致,携手发布Gluon深度学习库。意在提高所有开发者的机器学习技术的速度、灵活性和可访问性,而不考虑他们选择的深度学习框架。这项合作的第一个结果是产生了新的Gluon接口,它是Apache MXNet的一个开源码库,能够允许所有级别的开发人员对深度学习模型进行原型化(prototype)、构建和训练。这个接口极大地简化了创建深度学习模型的过程,同时又不原创 2017-12-29 20:18:52 · 1119 阅读 · 1 评论 -
Ubuntu循环登录解决
本文系转载,出处——【解决】Ubuntu安装NVIDIA驱动后桌面循环登录问题:http://blog.youkuaiyun.com/u010159842/article/details/54344683。今天配置深度开发学习环境,结果在Ubuntu装完NVIDIA驱动后就悲剧了,无限循环在登录界面。环境Ubuntu 14.04.5 GTX960m cuda8.0 + cudnn5.0转载 2017-11-25 16:36:13 · 3381 阅读 · 0 评论 -
Ubuntu下Ananconda安装Caffe2
Python2.7和Python3.6下都可以,不过只是cpu版本,只限于Mac和Ubuntu平台下: conda install -c caffe2 caffe2原创 2018-02-01 11:32:08 · 3158 阅读 · 1 评论 -
Ubuntu14.04下载cuda-8.0安装包
wget http://172.20.216.118/files/11540000001C0F53/developer.download.nvidia.com/compute/cuda/8.0/secure/Prod2/local_installers/cuda.deb原创 2018-04-22 19:24:17 · 681 阅读 · 1 评论 -
Windows下安装PyTorch0.4.0
本文系转载,出处:https://blog.youkuaiyun.com/sunqiande88/article/details/80085569。PyTorch简介在2017年1月18日,facebook下的Torch7团队宣布PyTorch开源后就引来了剧烈的反响。PyTorch 是 Torch 在 Python 上的衍生版本。Torch 是一个使用 Lua 语言的神经网络库, Torch 很...原创 2018-04-27 09:21:52 · 52320 阅读 · 27 评论 -
win10下导入keras/tensorflow报错
我在win10 Python3.5.2安装keras,然后导入时报错如下:Python 3.5.2 |Continuum Analytics, Inc.| (default, Jul 5 2016, 11:41:13) [MSC v.1900 64 bit (AMD64)] on win32Type "help", "copyright", "credits" or "license" for原创 2017-05-28 22:06:19 · 24711 阅读 · 2 评论