joyjoy01-CSDN博客

原创 pytorch模型转tensorrt时空洞卷积导致的输出结果错误解决办法

最近将一个带有空洞卷积的pytorch模型转为tensorrt加速，在转换过程中，出现pytorch->onnx->tensorrt，pytorch->tensorrt两种方式均能完成转换，但是转换后的模型输出结果不对，跟pytorch模型输出结果差别很大。tensorrt 空洞卷积层后面接个ElementWise层，ElementWise层输出结果跟pytorch模型中卷积层的求和结果是不一样的。模型结构：block{conv1=conv(input,kernel

2021-05-11 17:14:24 1245 3

原创 windows cuda the launch timed out and was terminated 解决方案

在windows下cuda函数运行时间超过5s时，windows会强制结束程序，导致输出结果不对。解决方式可分为三种：1、优化代码，减少程序的运行时间。2、修改注册表： 1)、左下角搜索->regedit 2)、找到HKEY_LOCAL_MACHINE->SYSTEM->CurrentControlSet->control->GraphicsDrivers 3)、右键GraphicsDrivers->新建->DWORD（32 位）,命...

2020-11-20 17:46:23 4416

原创 Caffe GPU编译报nvcc fatal : Unsupported gpu architecture 'compute_**' 错误解决方法

在cuda安装成功，配置好cudnn的条件下，编译Caffe GPU时生成工程时会报nvcc fatal : Unsupported gpu architecture 'compute_**' 错误，解决办法两步：1、.修改nvcc.hpp文件，注释最后三行。2、修改caffe-windows/Cuda.camke,将第7行修改为set(caffe_known_gpu_archs "6...

2019-05-10 17:26:14 6679 2

原创 tensorflow c++ 接口跑yolov3同darknet跑yolov3时间对比

显卡：GRID P-40tensorflow版本：1.5yolov3:416模型TensorFlow 1157张图像平均耗时99.1019毫秒。darknet 1157张图像平均耗时29.79毫秒。感觉用tensorflow做视频的目标检测不是很好啊。...

2019-03-01 18:38:33 2276 3

原创利用dl4j识别图像颜色

import org.datavec.api.records.reader.RecordReader;import org.datavec.api.records.reader.impl.csv.CSVRecordReader;import org.datavec.api.split.FileSplit;import org.datavec.api.util.ClassPathResourc...

2018-05-18 14:32:35 1686

原创图像颜色kmean聚类

import org.bytedeco.javacpp.Pointer;import org.nd4j.linalg.api.buffer.DataBuffer;import org.nd4j.linalg.api.ndarray.INDArray;import org.nd4j.linalg.api.ops.impl.indexaccum.IAMax;import org.nd4j....

2018-05-18 14:26:27 1299

原创提取车牌轮廓

public class Test { public static void main(String[] argv) throws InterruptedException { //gray Mat image = ImageUtil.readImage("E:/DataExamples/car2.jpg"); Mat filter ...

2018-05-18 14:21:44 959

原创 tensorflow手写字体识别，入门示例

此篇包含了手写字体识别的全连接神经网络和卷积神经网络模型建模，模型保存成pb格式，以及使用tensorflow迁移学习的方式。#!/usr/bin/python# -*- coding:utf-8 -*-import input_data as inputimport tensorflow as tffrom tensorflow.python.framework import graph_uti...

2018-05-18 14:02:47 1683

原创 tensorflow迁移学习示例

这种迁移学习并不能将两个模型合成一个，用起来很不方便# -*- coding:utf-8 -*-import input_data as inputimport tensorflow as tffrom tensorflow.python.framework import graph_utilfrom tensorflow.python.platform import gfiledef weigh...

2018-05-18 13:50:35 1492 1

原创 tensorflow迁移学习

模型是我自己保存的手写字体模型#!/usr/bin/python# -*- coding:utf-8 -*-import input_data as inputimport tensorflow as tffrom tensorflow.python.framework import graph_utilfrom tensorflow.python.platform import gfileimp...

2018-05-18 11:26:16 272

原创 tensorflow 训练mnist,及模型保存成pb文件

#!/usr/bin/python# -*- coding:utf-8 -*-import input_data as inputimport tensorflow as tffrom tensorflow.python.framework import graph_util#全连接神经网络训练手写字体def testMnist(): mnist = input.read_data_sets...

2018-05-18 11:19:19 4240 1

原创关于Nd4jBackend$NoAvailableBackendException: 的异常问题

我的maven只引用这datavec-api ，deeplearning4j-core，nd4j-native-platform的依赖，在deeplearning4j 0.7版本没有问题。后来改成0.91版本就出现异常：Caused by: java.lang.RuntimeException: org.nd4j.linalg.factory.Nd4jBackend$NoAvailableBacke

2017-09-20 13:05:48 3176 1