![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
AI
DvLee1024
微信同号,更多内容请到博客:https://shop.hidavid.cn
展开
-
人工智能检测医学图像——子宫肌瘤B超图
医学图像/目标检测/B超图像识别/子宫肌瘤原创 2021-05-26 17:39:03 · 1028 阅读 · 0 评论 -
判断CUDA和CUDNN是否安装成功并正常使用
最近又捡起YOLOv3来练练手,检测医学B超图像。重新搭建环境,由于网速时快时慢,搭建起来相当痛苦,最终还是搭建完成了。下面分享下如何判断CUDA是否正常使用。正文一、判断安装情况首先是判断cuda是否安装成功。一般安装路径为/usr/local/cuda使用nvcc -v命令可以输出cuda版本然后是判断cudnn,这个库安装很简单,只需把cudnn的include和lib64里面的文件拷到cuda相应目录即可,所以判断是否安装的方式是,到cuda的include和lib64,用ls |原创 2020-10-28 21:47:03 · 30289 阅读 · 1 评论 -
制作YOLOv5的数据集
在这里推荐我常用的标注工具 VIA,用久了,很多转化脚本都是基于这个工具导出的格式,所以没用其他标注工具了。VIA可以在线标注,地址是:http://www.robots.ox.ac.uk/~vgg/software/via/via.html标注完数据后,保存项目,导出一份json下面代码是把json转换成YOLOv5的格式import osimport jsonanno_file_path = '' // json路径saveFolder = r'./txt'if not os.pat原创 2020-08-14 17:25:20 · 2258 阅读 · 1 评论 -
隔空控物——基于YOLOv5的手势识别
先占坑,有空补回来 隔空控物,用Yolov5训练的手势识别原创 2020-08-11 09:16:30 · 2356 阅读 · 3 评论 -
Windows环境搭建CenterNet
Github上搜了一下,发现CenterNet关注量最多的就是下面的实现https://github.com/xingyizhou/CenterNet搭建过程遇到好多问题,一一解答。这个issue分享的是在Windows下搭建CenterNethttps://github.com/xingyizhou/CenterNet/issues/7issue作者使用的环境是Pytorch 1.1 ...原创 2020-03-31 09:54:45 · 1845 阅读 · 6 评论 -
windows环境下安装CUDA(用GPU训练Tensorflow-YOLOv3)
今天终于搞到一台带1080独显的PC,之前一直用CPU训练,效率极其低下。训练tensorflow-yolov3这个网络模型,图集400张,4个分类,要训练十多个小时才能看到检出效果,现在可以尝试使用GPU训练了。之前对GPU训练还不太了解,以为要改代码,现在才知道,其实不需要的。关键就是搭建好cuda环境NVIDIA显卡驱动(https://www.nvidia.cn/Downloa...原创 2019-07-19 18:57:01 · 3860 阅读 · 1 评论 -
Yolo v3 学习笔记摘录
作为一名移动端开发人员,我觉得现在是入门深度学习的最佳时机,毕竟tensorflow也发展好几年了,Facebook也推出了pytorch,github上已经有很多开源的各种神经网络的源码,可以比较轻易的实现一些震撼自己的效果。之前在某公司的计算机视觉部工作,受到深度学习工程师的耳濡目染,离职后,自己才真正去尝试数据采集、标注、训练、移植Android端等步骤,算是草草入门了。下面分享几段学...原创 2019-07-11 16:00:05 · 465 阅读 · 1 评论 -
如何在PC端测试转换后的TFLite模型
上篇讲述如何把tensorflow模型转换成tflite模型,用于部署到移动端。这篇分享如何在PC端对tflite模型进行预测,测试模型是否可用首先,加载tflite模型,查看模型的输入输出import numpy as npimport tensorflow as tfimport cv2# Load TFLite model and allocate tensors.inter...原创 2019-06-30 11:57:04 · 5464 阅读 · 12 评论 -
如何把Tensorflow模型转换成TFLite模型
深度学习迅猛发展,目前已经可以移植到移动端使用了,TensorFlow推出的TensorFlow Lite就是一款把深度学习应用到移动端的框架技术。使用TensorFlowLite 需要tflite文件模型,这个模型可以由TensorFlow训练的模型转换而成。所以首先需要知道如何保存训练好的TensorFlow模型。一般有这几种保存形式:CheckpointsHDF5SavedM...原创 2019-06-30 11:24:05 · 8810 阅读 · 4 评论 -
如何标注mask用于图像分割模型训练(VIA标注semantic segmentation数据集的mask)
近几年深度学习发展非常迅猛,深度学习用于图像识别、分割等方面效果非常好,像mask rcnn这类的网络已经可以做到对象分割了(instance segmentation)。训练图像分割模型,需要标注大量mask图片,而网上找到的标注工具只能导出json文件,今天分享下如何把json文件转化成mask 图片。首先,使用VIA标注工具,标注物体轮廓,导出json文件。json文件里包括了图片中物体...原创 2019-06-25 11:14:35 · 17747 阅读 · 29 评论 -
使用OpenCV进行抠图并计算物体的最小外接边框
先占坑,后面会补上一篇对背景简单的图片进行抠图,背景分割,提取前景信息,并计算最小外接边框原创 2019-03-19 16:15:23 · 993 阅读 · 2 评论 -
快速入门深度学习——Tensorflow
环境搭建IDE使用PyCharmPython版本 3.6.8(3.7版本无法安装TensorFlow)代码测试TensorFlow环境安装成功from __future__ import absolute_import, division, print_functionimport tensorflow as tffrom tensorflow import keras# 输出T...原创 2019-02-13 18:03:34 · 359 阅读 · 0 评论 -
快速入门深度学习——跑通TensorFlow的MNIST图集
学编程都喜欢运行HelloWorld,而深度学习的HelloWorld便是跑通MNIST。MNIST是一个手写数字的图集库,图集非常精小,非常适合作为深度学习练手的图集。跑完下面的demo,会对TensorFlow训练模型和识别图片数字这个流程有个整体的概念。首先导入需要的库,能正常输出tensorflow的版本号表示环境没有问题。若环境存在问题,请先阅读上一篇文章。from __...原创 2019-02-15 16:37:46 · 652 阅读 · 1 评论