- 博客(7)
- 收藏
- 关注
原创 yolov2-----源码阅读【reorg层】
官方yolov2 结构如图reorg层的作用是将上一层卷积后的特征图38X38X64 拉伸成19X19X256,具体看代码: elif module_type == "reorg": stride = int(module["stride"]) #stride 为2 B,C,H,W = x.size() #B为bach大小暂设为32,c为通道数64 H,W都为38 x = ...
2019-07-24 19:20:38 1383 1
原创 python提取COCO,VOC数据集中特定的类
1.python提取COCO数据集中特定的类from pycocotools.coco import COCOimport osimport shutilfrom tqdm import tqdmimport skimage.io as ioimport matplotlib.pyplot as pltimport cv2from PIL import Image, ImageDr...
2019-07-24 17:21:40 11171 6
原创 论文阅读【模型剪枝(二)】
1. Learning Efficient Convolutional Networks through Network Slimming论文地址:https://arxiv.org/abs/1708.06519论文代码:https://github.com/talebolano/yolov3-network-slimming
2019-07-16 16:41:17 474 1
原创 【解决】新版cudnn Deb安装版本 找不到cudnn.h
最近发先NVIDIA官网更新了所有的cudnn安装包,之前都是tar.gz格式,执行:tar xvf cudnn-版本号.tgzcp -ar cuda/* /usr/local/cudasudo ldconfig现在跟新的都是ded安装包,ubuntu安装一定要下载箭头的cuDNN Library for Linux版本,解压安装和之前一样哈。如果安装了Deb版本,使用以下命令卸载...
2019-07-16 15:13:09 8760 4
原创 【解决】Check failed: status == CUDNN_STATUS_SUCCESS (8 vs. 0) CUDNN_STATUS_EXECUTIO
配置:2080TI,cuda9.1, cundnn7.5.0 ,Nvidia驱动430,Ubuntu16.04+内核4.10.0执行caffe出现问题:Check failed: status == CUDNN_STATUS_SUCCESS (8 vs. 0) CUDNN_STATUS_EXECUTIO编译darknet 取消cudnn可以通过,但是测试demo 时,没有结果,这就奇怪了好...
2019-07-16 14:52:38 8377 13
原创 论文阅读【模型剪枝(一)】
最近看了几篇关于剪枝的paper,汇总如下:1.Channel Pruning for Accelerating Very Deep Neural Networks论文地址:https://arxiv.org/abs/1707.06168代码地址: https://github.com/yihui-he/channel-pruning对于完成网络训练后的模型加速的方法一般有三种:优化实现(...
2019-07-15 15:39:31 783
原创 python 图像批量png转jpg格式
import osimport stringdirName = "D:your path\\" #最后要加双斜杠,不然会报错li=os.listdir(dirName)for filename in li: newname = filename newname = newname.split(".") if newname[-1]=="png": ...
2019-07-01 15:38:16 4564 5
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人