- 博客(4)
- 收藏
- 关注
原创 二阶段目标检测网络模型压缩网络剪枝
二阶段目标检测网络模型压缩网络剪枝由于网络上对于二阶段目标检测网络的网络剪枝资料较少,所以对剪枝过程进行了复现,本文以faster rcnn为例。使用的方法是network slimming,通道剪枝,通过对bn层gamma的分布分析,剪枝接近0的整个神经元通道。论文:Learning Efficient Convolutional Networks Through Network Slimming链接: https://arxiv.org/abs/1708.06519.剪枝原理其他的博客很多,在
2021-08-12 18:27:33 1218 5
原创 基于pytorch的目标检测数据增强(tensor数据流版本)
我的示例代码的dataloader中打包传入的是一个target字典,里面包括boxes和label,如果你们传入的是boxes和label,直接修改参数就行了,然后因为我传入的image和target都是经过torch的转换,数据格式是tensor,所以有一些转换格式的代码,然后图片shape是(c,h,w),随机概率设的是0.3,都按需要修改就行。随机缩放class randomScale(object): def __call__(self,image,target): #
2021-02-01 11:31:25 2009 10
原创 Python在windows下使用ffmpeg进行视频截取帧照片并保存到相应文件夹
Python在windows下使用ffmpeg进行视频截取帧照片ffmpeg的一些命令行是在终端使用的并且大多数是以sh文件作为脚本文件运行,在windows下运行linux脚本较为麻烦,所以我写了一个可以遍历文件夹的py脚本文件在windows下使用ffmpeg进行视频帧的截取图片,并创建在有完全相同的数据集目录。实际上也可以利用ffmpeg做其他操作包括录屏等等,可以自己去查询ffmpeg的使用方法。起因是我在跑C3D项目代码的时候需要对UCF-101数据集进行视频到帧图片的transform,但是
2020-09-16 09:06:17 4153 7
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人