《表4 目标检测网络裁剪前后效果对比》
在目标检测的实验中,本文从公开数据集COCO、VOC中提取了含有person类别的图片,并从中随机选取6 300张图片作为本次实验的数据集,其中5 000张作为训练集,1 300张作为测试集。训练时,采用pytorch框架,将训练集中的图片统一变为416像素×416像素大小输入检测网络进行训练。训练完成后,先对该模型进行通道剪枝,接着对其稀疏化训练并层剪枝,剪枝完成后对网络再一次训练。最后在装有Ubuntu16.04操作系统的笔记本电脑上对网络模型进行测试,测试结果如表4所示。
图表编号 | XD00208476000 严禁用于非法目的 |
---|---|
绘制时间 | 2021.01.01 |
作者 | 宗佳平、吴妍、陈建强、张琳娜、张悦、岑翼刚 |
绘制单位 | 北京交通大学信息科学研究所、贵阳市公安司法鉴定中心、贵州大学机械工程学院、贵州大学机械工程学院、北京交通大学信息科学研究所、北京交通大学信息科学研究所 |
更多格式 | 高清、无水印(增值服务) |