深度学习
文章平均质量分 77
berrymusic
努力的小爬虫。
展开
-
pytorch实现并训练MobileNetV3
1.MobileNetV3MobileNet是Google公司推出的轻量化系列网络,用以在移动平台上进行神经网络的部署和应用。2019年,Google发布了第三代MobileNet,即MobileNetV3。在MobileNet系列的精度和计算量上都达到了新的state-of-art,以下简单回顾一下三代MobileNet的主要特性:MobileNetV1:MobileNetV1的主要思想是将普通的卷积操作分解为两步,先做一次仅卷积,不求和的“深度卷积”操作,再使用1*1的“点卷积”对深度卷积得到的多原创 2020-07-16 10:52:41 · 16534 阅读 · 23 评论 -
pytorch训练自己的YOLOv5目标检测器(自定义数据集训练)
1.YOLOv5YOLO系列可以说是单机目标检测框架中的潮流前线了。YOLO系列的原作者虽然放弃了YOLO系列的开发,但是俄罗斯的开发者Alexey接过了YOLO系列的大旗,今年四月份正式推出了YOLOv4,并开源了代码,论文也发了。YOLOv4是基于darknet平台的,使用官方开源代码需要安装Visual Studio并使用Cmake来编译,inference和train起来都很不方便,so,在github上曾复现YOLOv3的大神ultralytics综合了YOLO系列的各种长处,推出了YOLOv原创 2020-06-24 19:01:29 · 18255 阅读 · 61 评论 -
cuda10.1安装tensorflow-gpu1.x版本时ImportError: DLL load failed: 找不到指定的模块问题解决
问题描述最近在跑语义分割的经典网络,SegNet,PSPNet等,在github上发现2017年左右的论文代码都是基于tensorflow1.x的,于是使用anaconda新配置了一个虚拟环境,预期使用keras2.1.5+tensorflow1.13.2,前期的keras等基本包使用pip均可以正常安装,可是tensorflow-gpu在安装后无法导入,提示ImportError: DLL load failed: Failed to load the native TensorFlow runtime原创 2020-06-15 18:42:03 · 601 阅读 · 0 评论 -
语义分割网络FCN的思想概括与调试运行
1.FCN简介FCN是语义分割的经典论文,其核心思想就是把基本的分类网络作为backbone进行改造,将全连接层改造为卷积层以适应对密集像素点进行分类。因此,传统的分类网络输出的是概率向量,在概率向量中取最大值作为最终结果。而FCN网络则输出的是低分辨率的一个feature map(论文中称之为heatmap,即热图),如下如图所示:这个heatmap对应着图片中的高级信息,虽然人无法再理解这张图的内容,但神经网络却可以理解这个图,并基于这个heatmap来完成逐像素点的分类任务。如果仅将原始分类网络原创 2020-06-02 10:39:03 · 719 阅读 · 0 评论 -
Pytorch下Retinanet的代码调试
最近在复现经典cv论文的网络结构,经典的AlexNet,VGG等网络由于基本都是同源的。基本只是深度和预处理的代码不同,因此用Pytorch搭建起来很容易,到了RetinaNet,由于其将多个网络融合,代码和实验量较大(RetinaNet论文的实验量吓到我了,真、实验狂魔)复现起来较困难,因此选择了取github上下载大佬的代码来用。此帖记录了跑代码的过程和全程遇到问题的解决方案。一、项目链接...原创 2020-04-12 14:12:18 · 2869 阅读 · 6 评论