- 博客(63)
- 资源 (10)
- 收藏
- 关注
原创 OnnxRuntime C++不能使用CUDA的问题
问题描述:当使用onnxruntime进行推理的时候发现使用的是cpu,而不是GPU解决办法:有可能是头文件和设置的问题step1:头文件添加:#include <onnxruntime_c_api.h>#include <cuda_provider_factory.h>step2:session设置:Ort::SessionOptions session_options;session_options.SetInterOpNumThreads(1
2022-05-31 16:40:36 3285
原创 神经网络参数量计算-UNet3D为例
模型:import torchimport torch.nn as nnimport torch.nn.functional as Fimport numpy as npimport osclass UNet3D(nn.Module): """ Baseline model for pulmonary airway segmentation """ def __init__(self, in_channels=1, out_channels=1, coord=True): "
2022-04-15 11:27:04 2358 2
原创 论文解读:AI-assisted CT imaging analysis for COVID-19
论文解读:AI-assisted CT imaging analysis for COVID-19 screening:Building and deploying a medical AI system in four weeks使用二阶段分割,先使用分割网络分割出有病变的肺部组织,然后利用分类网络用于辨别是否是CONVID.文章中分割网络实验了好几种,最终使用的是3D UNet++分类网络是用的是Attention Resnet50这个AI辅助系统之所以能够快速的部署,原因有几点:
2022-03-21 18:43:29 751
原创 pytorch安装教程,使用国内源
1.cuda+cudnn+ 8.2.42.pytorchpip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio===0.10.1+cu113 -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com --default-timeout=10000 -f https://download.pytorch.org/whl/
2021-12-21 14:35:03 4094
原创 windows pyinstaller打包pytorch程序到exe文件
主要内容利用pyinstaller将pytorch代码打包成exe其实含有比较多大的坑,这篇文章主要记录最近在项目中打包exe遇到的一些问题,由于问题主要是opencv和pytorch带来的,解决的办法和minist的pytorch程序差不多,这个minist程序我已经测试过,能够打包成功,并在没有python环境的电脑上成功运行:https://github.com/1240117300/MINIST。出坑技巧一:安装匹配的包版本这是虚拟环境中的安装包及对应的版本号,这张表就能差不多解决问题了
2021-11-12 14:36:07 4504 1
原创 ITK 5.2.0 编译 和 win64 dll下载
参考网址:https://vtk.org/Wiki/VTK/Configure_and_Build安装QT5.12通过https://doc.qt.io/qt-5/gettingstarted.html下载offline installer安装VS2019下载CMakehttps://cmake.org/download/下载VTKhttps://vtk.org/download/https://vtk.org/files/release/8.2/VTK-8.2.0.zip
2021-07-20 06:08:47 556
原创 protocbuf如何手动编译-ubuntu
protocbuf如何手动编译-ubuntu有时候经常会遇到需要protocbuf版本升级的问题,所以apt的形式有时候无法满足有求,这里介绍如何通过手动安装protobuf第一步,下载protocbuf相应的版本https://github.com/protocolbuffers/protobuf第二步,编译安装autoconfsudo apt-get install autoconf开始编译cd protobufbash autogen.sh./configuremakemake
2021-05-13 10:52:50 215
原创 TCP/IP socket通讯
TCP IP协议为什么分层:实现解耦,层和层之间接口稳定就先,内部可以自己实现三次握手,资源的开辟,就是建立连接的过程三次握手之后,才能数据传输, 然后断开,四次分手什么是四次分手什么是socket,socket是四元组(ip+port ip+port)portd端口号的取值范围65535,我们一般用1000以后的一个服务器可以建立多个客户端与他的连接,四次分手什么是服务端:tcpdum -nn -i eth0 port 80mac地址在变,ip不变...
2021-04-07 11:50:53 183
原创 ffmpeg4.3 ndk编译,andorid调用ffmepg解码rtsp 坑坑总结
ffmpeg4.3 nkd编译编译过程:1.下载源码我编译的是ffmpeg4.3版本git clone https://git.ffmpeg.org/ffmpeg.git ffmpeg2.改写configure文件找到注释掉的这几行,用下面的替换#SLIBNAME_WITH_MAJOR='$(SLIBNAME).$(LIBMAJOR)'#LIB_INSTALL_EXTRA_CMD='$$(RANLIB) "$(LIBDIR)/$(LIBNAME)"'#SLIB_INSTALL_NAME
2021-03-24 15:25:33 304
原创 Windows下安装 pycocotools
Windows下安装 pycocotools亲测可行2020-12-16日测试原版的pycocotools(不支持windows)的Github项目地址:https://github.com/cocodataset/cocoapi支持windows的改写版本的Github项目地址:https://github.com/philferriere/cocoapi安装步骤:安装pycocotools首先需要安装Cython,在相应的conda环境下输入:pip install cython
2020-12-16 11:45:17 270
原创 远程连接服务器,并开启pytcharm遇到的坑坑
需要传图形界面,远程linux服务器,传图形界面,需要用到-X选项ssh -X [email protected]有时候pycharm运行不起来,有可能是由于配置了conda,而且没有默认启动,可以通过如下命令解决:source ~/.bashrc...
2020-12-09 16:37:38 191
原创 mysql 远程连接连不上的解决办法
找到/etc/mysql/mysal.conf.d的文件,找到bind-address = 127.0.0.1然后注释掉重启mysqlservice mysql restart
2020-12-01 20:26:35 219
原创 mmdetection mask-rcnn 训练步骤
mmdetection mask-rcnn 训练步骤1、准备数据集,coco标注格式2、修改匹配的训练参数文件configs/mask_rcnn_r50_fpn_1x_coco.py这里配置了模型文件,数据配置文件以及调度设置等configs/base/datasets/coco_instance.py修改data_root,train、val、test的ann_file、img_prefix、img_scaleconfigus/base/models/mask_rcnn_r50_fpn
2020-11-20 14:11:52 6717 3
原创 制作TEC数据集的步骤
数据集制作步骤:第一步将元器件检测结果裁剪成独立的元器件图片1.crop_roi_from_detect_result.pyfolder_path中存放存放主板和主板对应的元器件输出文件,这个脚本是将检测的区域截取出来,并按照主板名称分别保存2.match_high_low_lumi.py这个python文件是用于由低曝光的图片的检测结果,同时截取低曝光和高曝光图片的元器件图片,并分别保存.需要对低曝光和高曝光图像进行匹配,才能抠出对应的图片.if __name__ == '__main__
2020-11-20 10:11:34 666
原创 python比较两个字符串有什么不一样的地方
使用方法:difflib.Differ().compare()a = "mongodb://asdfadsf:[email protected]:27107/user"b = "mongodb://asdfadsf:[email protected]:27017/user"from difflib import Differd = Differ()diff = d.compare(a.splitlines(), b.splitlines())print('\n'.join(list(d
2020-11-13 20:44:58 1876
原创 TEC_OCR docker服务器搭建步骤
TEC_OCR server 环境配置** 第一步 下载tec_ocr需要的镜像文件 **参照mmdetect2.3的dockerfile,下载镜像ARG PYTORCH="1.5"ARG CUDA="10.1"ARG CUDNN="7"FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-develENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0+PTX"ENV TORCH_NVCC_FLAG
2020-10-21 14:55:13 736
原创 Tensorrt自定义算子实现步骤
这篇文章主要是用于记录tensorrt自定义算子的过程.这里采用的是torch2trt直接转换的方式.第一步在模型中造一个自定义的算子:import torchfrom torch.nn.modules import Moduledef do_custom_Relu(x): return torch.clamp(x,min=0.0)def do_custom_op_shift(x,bias): x = x.cpu().detach().numpy() x = x + b
2020-09-21 17:49:00 5741 1
原创 ubuntu 18.04升级gcc 7.5步骤
系统自带的gcc是5.4版本的,工作中需要更高版本的gcc,这里介绍的是升级到7.5.0的方法:可以到这个链接下载gcc的源文件gcc-7.5.0.tar.gz下载地址http://mirrors.nju.edu.cn/gnu/gcc/gcc-7.5.0/下载其他依赖文件gmp-6.1.0.tar.bz2mpc-1.0.3.tar.gzmpfr-3.1.4.tar.bz2isl-0.16.1.tar.bz2分别解压,然后分别执行如下操作``./configuremake -j2sud
2020-09-17 16:44:20 6759 2
原创 samba在ubuntu和ubuntu之间创建共享文件夹
samba在ubuntu和ubuntu之间创建共享文件夹** 示例 **sever上smb.conf文件末尾添加[MyShare]comment = shares for userspath = /home/u1/mikeyna/MySharebrowseable = yespublic = nowrite list = u1client 端的etc/fstab末尾添加smb//10.5.11.90/MyShare /home/mikeyna/Data/MyShare/
2020-08-11 11:04:17 673
原创 MMDetect2.0模型下载地址
MMDetect2.0模型下载地址MMDetect2.0运行demo的时候,会遇到跑完前向之后,什么结果也没有画出来,原因是模型不匹配模型可以在如下网址查询https://open-mmlab.s3.ap-northeast-2.amazonaws.com/举个例子https://open-mmlab.s3.ap-northeast-2.amazonaws.com/加上后面的相对路径即可mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_caffe_fpn_
2020-07-10 19:04:56 450 1
原创 Tensorrt环境搭建
Tensorrt环境搭建环境安装python 3.6TensorRT 7.0.0.1https://developer.nvidia.com/nvidia-tensorrt-7x-downloadcd /path/to/TensorRT7.0.0.1pip install tensorrt-7.0.0.11-cp36-none-linux_x86_64.whlpytorch 1.6根据机器需求,下载匹配的pytorchcudnn 7.6https://developer.nvidia.
2020-07-01 15:20:24 977
原创 MNN windows 编译
@MNN Windows编译步骤参考链接https://www.yuque.com/mnn/cn/build_windowshttps://blog.csdn.net/ouyangfushu/article/details/96476245https://blog.csdn.net/jacke121/article/details/106403303步骤git clone https://github.com/alibaba/MNN.git打开x86_64 Cross Tools Comm
2020-06-30 12:26:12 1305
转载 如何以root权限启动应用
sudo chown root appsudo chmod u+s app为了使用qtcreator root权限调试:sudo chown root /usr/bin/gdbsudo chmod u+s /usr/bin/gdb
2020-05-08 18:45:49 1094
原创 linux中如何获取设备的唯一标识
1.一般使用磁盘的uuidsudo blkid2.使用dmi信息如序列号: sudo dmidecode -s system-serial-number但是,有些平板电脑上并没有这个序列号
2020-05-08 18:42:29 7194
原创 c++中将命令行中的输出到字符流
std::string GetStdoutFromCommand(std::string cmd){ std::string data; FILE *stream; const int max_buffer=256; char buffer[max_buffer]; cmd.append(" 2>&1"); stream = popen(cmd.c_str(), "r"); if(stream) { whil.
2020-05-08 18:39:29 447
原创 python写的一个选票系统
# -*- coding: utf-8 -*-"""Spyder EditorThis is a temporary script file."""# -*- coding: utf-8 -*-"""Created on Fri Nov 12 13:54:22 2019@author: mikey"""import numpy as np def count_nam...
2019-11-12 19:31:29 2020
转载 WIN10 + Tensorflow1.12 + Cmake编译 + Bazel编译
文章转自:https://www.wandouip.com/t5i38311/几篇值得参考的博文:[1]https://blog.csdn.net/robothn/article/details/86157724[2]https://blog.csdn.net/yz2zcx/article/details/83106669[3]https://blog.csdn....
2019-10-14 16:41:58 1223 1
原创 Tensorflow C++接口内存泄漏 使用tcmalloc方法
ubuntu 16.04上使用TCMalloc1.sudo apt-get install google-perftools2.在CMakeList.txt中连接动态库target_link_libraries(smartkit4-svs /usr/lib/libtcmaloc.so.4)参考网址:https://github.com/tensorflow/tensorf...
2019-08-12 18:45:45 818
原创 C++图片等比例缩放到固定大小
int minHW = min(img.rows, img.cols); float ratio = max(_input_height, _input_width) / (float)minHW; int newH = max(_input_height, (int)(img.rows * ratio)); int ne...
2019-07-24 16:09:38 3529
gcc-7.5-安装套件.zip
2020-09-17
模拟磁盘扇区管理程序
2008-12-31
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人