深度学习
文章平均质量分 69
机器人学渣
Freedom of thoughts.
展开
-
C++ 上用 ONNXruntime 部署自己的模型
利用C++ ONNXruntime部署自己的模型,这里用Keras搭建好的一个网络模型来举例,转换为onnx的文件,在C++上进行部署,另外可以利用tensorRT加速。目录一、模型的准备二、配置ONNXruntime三、模型的部署1. 模型的初始化设置2. 构建推理构建推理函数computPoseDNN()步骤:函数具体代码:四、应用参考一、模型的准备搭建网络模型训练:tensorflow keras 搭建相机位姿估计网络–例网络的输入输出为:网络的输入: [image_ref , ima原创 2022-01-21 17:56:53 · 18447 阅读 · 11 评论 -
tensorflow keras 搭建相机位姿估计网络--例
主要展示keras 的使用方法,搭建网络模型的步骤。目录概要一、数据集的处理二、网络模型的搭建三、模型的训练四、加载保存的模型概要这个网络是利用两帧图像,预测相机的位姿变换矩阵,总共6个自由度网络的输入: [image_ref , image_cur]网络的输出: [tx , ty , tz , roll , pitch , yaw] 一、数据集的处理数据集的格式:# image_ref image_cur tx ty tz roll(x) pitch(原创 2022-01-21 15:40:32 · 1400 阅读 · 1 评论 -
tensorflow 2.5.0 ( keras )搭建wgan-gp 和 div
使用tensorflow 2.5.0搭建wgan网络:代码:import argparsefrom inspect import classify_class_attrsimport tensorflow as tfimport tensorflow.keras as Kimport numpy as npimport cv2 as cvimport osimport time#采用静态图的形式,可关闭急切模式#tf.compat.v1.disable_eager_execution原创 2022-01-05 17:31:51 · 1715 阅读 · 2 评论 -
tensorflow Keras的搭建训练步骤
keras搭建训练模型步骤原创 2021-12-28 18:19:56 · 1933 阅读 · 3 评论 -
在C++上利用onnxruntime (CUDA)和 opencv 部署模型onnx
概述将得到的模型转化为onnx模型,加载到c++中运行,来完成模型的部署,下载并安装onnxruntime;CMakeLists.txt:cmake_minimum_required(VERSION 2.8)project(test)#使用clang++编译器set(CMAKE_CXX_COMPILER clang++)set(CMAKE_BUILD_TYPE "Release")set(CMAKE_INCLUDE_CURRENT_DIR ON)#find the opencv and原创 2021-10-30 15:55:36 · 12327 阅读 · 12 评论 -
在ubuntu 20.04 上 安装 onnxruntime CUDA
onnxruntime 可以加速模型的部署,同时可以跨框架,例如源码项目位置:https://github.com/microsoft/onnxruntime里面有一些依赖,这里直接打包,提供百度网盘的下载:链接:https://pan.baidu.com/s/1sTryvfX0sLKIoHm8aMm4Sg 提取码:be9o解压后:...原创 2021-10-29 15:17:17 · 3660 阅读 · 4 评论 -
MASK RCNN在tensorflow 2.5中运行
原mask rcnn的tf版本是1.13,这里提供tf2.5的mask rcnn源代码:源github地址:https://github.com/matterport/Mask_RCNNtf2.5版本 github地址:https://github.com/zouyuelin/MASK_RCNN_2.5.0目录一、安装依赖二、数据集配置三、修改train.py四、训练五、测试一、安装依赖pip install -r requirements.txt -i https://pypi.tuna.t原创 2021-10-10 14:05:28 · 2948 阅读 · 39 评论 -
利用tensorFlow api 识别手术器械
目录前言一、下载coco-trained models:二、数据集的处理三、修改pipeline.config文件四、训练五、保存节点pb六、生成pbtxt文件,opencv调用需要七、测试图片数据集来源:18岁NIPS Workshop一作,用目标检测评估手术技能点Tensorflow-model:(选中master,点击tag选择和自己tensorflow适配的版本)https://github.com/tensorflow/models/包含手术器械数据集及tensorflow-model所需原创 2021-06-07 12:58:16 · 1041 阅读 · 1 评论 -
keras搭建wgan-gp和wgan-div,可生成图像
文章目录一、前言二、wgan-Gp原理三、wgan-Div原理四、代码结构设计过程4.1.生成tfrecord4.2.设计残差网络结构搭建resBlock模块4.3 搭建generate网络:4.4.搭建discriminator网络:4.5.定义网络的损失函数:a.首先得到判别网络和生成网络:b.定义训练判别网络gan_train_d:c.定义训练判别网络gan_train_g:4.6.定义训练网络循环体:4.7.其他函数:plot()五、代码及训练结果5.1.运行代码5.2.使用wgan-Gp生成训练.原创 2020-12-21 00:18:59 · 3487 阅读 · 18 评论 -
tf.slim构建vgg16和resnet网络实现图像分类,亲测准确率99%
tf.slim构建vgg16和resnet网络实现图像分类,亲测准确率99%文章目录前言一、大致步骤二、具体实施步骤1.创建tfrecords文件2.搭建前向传播网络forward()2.1 resnet2.2 vggnet2.3 simplenet2.4 forward()3.搭建反向传播网络backward()3.1 placeholder占位3.2 损失函数3.3 建立评价函数accuracy3.4 学习率的设定(指数下降)3.5 建立优化器:Adam优化器3.6 读取数据:训练原创 2020-11-30 20:46:53 · 1571 阅读 · 2 评论 -
tensorflow.keras搭建gan神经网络,可直接运行
tensorflow.keras搭建gan神经网络,可直接运行文章目录tensorflow.keras搭建gan神经网络,可直接运行前言一、tf.keras搭建gan网络大致步骤二、使用步骤1.制作tfrecords数据集2.读入数据3.搭建gan网络a.搭建generator网络b.搭建discriminator网络c.整合generator,discriminator网络为gan网络4.complie编译(建立loss和optimizer优化器)5.训练网络(建立循环)6.保存网络7.完整的gans.原创 2020-11-27 09:46:53 · 1542 阅读 · 1 评论