- 博客(32)
- 收藏
- 关注
原创 c++ load pytorch 的数据转换
tensor、numpy、vector转换python中:**numpy -> tensor**: `torch.from_numpy(ndarray)` **tensor -> numpy**: `tensor.numpy()`c++中:**array -> tensor**: `torch::tensor(at::ArrayRef<float&...
2019-05-24 10:42:13
9606
16
原创 如何在c++中使用pytorch模型
同步知乎页:https://zhuanlan.zhihu.com/p/66707105该过程有两步:第一:在python中导出pytorch模型第二:在c++中加载并执行第一步:convert Pytorch func/model to Torch Script有2中方法可以完成该任务。一、 torch.jit.trace function def foo...
2019-05-23 10:33:31
1624
3
原创 CMakeLists常用写法
基本结构cmake_minimum_required(VERSION 3.0)project(DEMO) add_executable(run main.cpp) 添加头文件目录目录结构:+root +build +lib +heads -func.hpp +src -func.cpp -main.cpp -CMakeLists.txt下文所有的cma...
2019-05-18 15:58:35
725
原创 tensorflow serving 配置
一、由checkpoint导出pb文件如下是deeplab v3导出模型的代码,可参考'signature_def_map' dict的key名字要记住,需要与后边统一。import osimport tensorflow as tffrom tensorflow.python.tools import freeze_graphfrom deeplab import comm...
2018-07-10 14:28:45
1963
原创 语义分割数据集汇总
cityscapes A new large-scale dataset that contains a diverse set of stereo video sequences recorded in street scenes from 50 different cities, with high quality pixel-level annotations of 5 000 frame...
2018-04-25 14:02:43
3700
原创 MobileNet V2 论文笔记
论文:Inverted Residuals and Linear Bottlenecks: Mobile Networks for Classification, Detection and Segmentation文章在residual net和MobileNet V1的基础上,提出MobileNet V2模型,一方面保证准确性,另一方面大幅的减少multiply-adds(MAdd)的计算量,从
2018-03-01 11:38:23
9063
3
原创 风格迁移系列-Image Style Transfer Using Convolutional Neural Networks
论文Image Style Transfer Using Convolutional Neural Networks概述及相关工作图像风格迁移任务,可以概括为把一张图像的纹理信息加到另一种图像上,而保持该图像的语义内容(比如,图像是一张狗的照片)不变。变换的难点在于如何明确的表征图像的语义信息,从而允许把图像的‘content’与‘style’二者分割开来。 风格迁移有时可以认为是纹理迁移。而纹理
2018-01-03 11:00:09
2193
原创 Image style transfer
论文Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization图像风格变换的任务是,把一种风格域S1的图像转换成另一种风格域S2的图像。风格变换的同时保存图像的内容不变。GAN火起来之后,有一些用GAN做图像风格变换的应用,比如Cycle GAN/Dual GAN/Disco GAN。他们的一般结构是使用两对G
2017-11-26 18:56:52
1791
原创 PROGRESSIVE GROWING OF GANs for IMPROVED QUALITY, STABILITY AND VARIATION
PROGRESSIVE GROWING OF GANs for IMPROVED QUALITY, STABILITY AND VARIATIONNVIDIA深度视觉实验室提出的一种训练GAN的方法。
2017-11-17 11:21:31
3914
1
原创 正则表达式在文本过滤中的使用
正则表达式在文本过滤中的使用这个系列简单记录学习正则表达式的历程,结合在grep awk sed 中的使用,增强印象。
2016-05-29 11:14:40
704
原创 Check failed: error == cudaSuccess (8 vs. 0) invalid device function
运行faster-rcnn做测试时可能出现的错误 google找到了答案,是nms的问题,设置nms不使用gpu即可,但是为什么这样还不是很清楚。 解决: in $FCN_ROOT/lib/fast_rcnn/config.py set __C.USE_GPU_NMS = False
2016-05-28 11:12:03
3750
原创 使用py-faster-rcnn训练自己的数据
使用py-faster-rcnn训练自己的数据faster-rcnn原本使用voc2007数据集训练,可识别20类目标。但是我们的应用场景,可能只需要2、3类。
2016-05-28 11:02:52
3993
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人