随笔分类 -  深度学习

上一页 1 2 3 4 5
mask生成平行框
摘要：在Mask Rcnn中传给模型训练的是物体的mask图，但训练中会利用到包含物体的平行框，查看代码后发现gt_box是由mask生成的，其主要思路是找出平行框左上角和右下角的坐标。mask图是一个二值图，利用numpy库的where()和any()函数就可以找出$x_1$和$x_2$以及$y_1$和 阅读全文
posted @ 2020-04-05 22:29 半夜打老虎 阅读(816) 评论(1) 推荐(0)
残差网络---ResNet
摘要：目录 一、残差块(Residual Block) 二、 残差网络为什么有用 三、ResNet网络结构 四、代码实现 ——————————————————————————————————————————————————————————————————————————————— 论文：https://a 阅读全文
posted @ 2020-03-21 23:09 半夜打老虎 阅读(3230) 评论(0) 推荐(0)
模型Mask_Rcnn训练错误：RuntimeError: It looks like you are subclassing 'Model'
摘要：一、问题： 最近在keras下利用两个GPU训练Mask_Rcnn模型的时候出现以下错误。 error:RuntimeError: It looks like you are subclassing 'Model' and you forgot to call 'super(YourClass, s 阅读全文
posted @ 2020-03-20 18:06 半夜打老虎 阅读(2065) 评论(0) 推荐(0)
模型训练错误：Failed to get convolution algorithm，This is probably because cuDNN failed to initialize.
摘要：一、问题： 最近在利用keras训练模型的时候出现以下错误，开始一看以为是cuDNN的版本不对，更换后还是没有效果，后面在网上查找后发现是显存的问题，更改为动态分配内存就可以了。 二、解决方法： 在训练的脚本开头添加以下代码进行动态分配内存，然后重新运行就可以。 import tensorflow 阅读全文
posted @ 2020-03-01 09:35 半夜打老虎 阅读(12917) 评论(0) 推荐(1)
MobileNet系列---mobileNetV2
摘要：最近在利用SSD检测物体时，由于实际项目要求，需要对模型进行轻量化，所以考虑利用轻量网络替换原本的骨架VGG16，查找一些资料后最终采用了google开源的mobileNetV2。这里对学习mobileNet系列的过程做一些总结。mobileNetV1是由google在2017年发布的一个轻量级深度 阅读全文
posted @ 2020-02-22 21:39 半夜打老虎 阅读(10281) 评论(0) 推荐(2)
【数据预处理】：图像去均值：image mean 和 pixel mean
摘要：原文链接：https://blog.csdn.net/weixin_37251044/article/details/81157344 1.什么是均值？ 对于每帧图像来说，均值分为两种：image mean 和 pixel mean。 image mean： 简单的说，读入一张彩色图像，假设是(NN 阅读全文
posted @ 2020-02-20 19:59 半夜打老虎 阅读(2340) 评论(0) 推荐(0)
MobileNet系列---mobileNetV1
摘要：最近在利用SSD检测物体时，由于实际项目要求，需要对模型进行轻量化，所以考虑利用轻量网络替换原本的骨架VGG16，查找一些资料后最终采用了google开源的mobileNetV2。这里对学习mobileNet系列的过程做一些总结。mobileNetV1是由google在2017年发布的一个轻量级深度 阅读全文
posted @ 2020-02-16 22:33 半夜打老虎 阅读(6548) 评论(2) 推荐(1)

上一页 1 2 3 4 5