yolov3-channel-and-layer-pruning icon indicating copy to clipboard operation
yolov3-channel-and-layer-pruning copied to clipboard

yoloV4-tiny同时剪层和通道后,检测耗时与原始模型基本相同?

Open basailuonaaa opened this issue 4 years ago • 3 comments

测试环境:Ubuntu16.04, RTX2080Ti(11G),TensorRT-6.0.1.5, Cuda 10.0, Cudnn 7.6.1; YoloV4 模型为:512*512 tiny模型, 7类;

同时剪层和通道,shortcuts 12 ,global_percent 0.5,迭代300次后进行模型微调训练,模型大小也由原来的23.6M降为了8.6M,但是处理耗时基本没变。 耗时计算方法:处理1514张1080P图像,计算平均耗时; yoloV4-tiny原始模型耗时:14.4738毫秒; 压缩与剪枝后的模型耗时:13.7837毫秒;

两个模型的平均处理耗时变化很小,这是什么原因呢?

basailuonaaa avatar Sep 15 '20 08:09 basailuonaaa

通道剪枝(减掉了冗余参数) 增速不明显 减层就明显但精度就差了很多

zbyuan avatar Sep 16 '20 09:09 zbyuan

减层也不明显啊

deeplearningers avatar Nov 19 '20 06:11 deeplearningers

@basailuonaaa 你好,我刚接触这方面的学习。我发现此项目基于yolov3,想请教一下若想用yolov4-tiny,该如何实现呢?感谢!

sticktotheend avatar Mar 05 '24 02:03 sticktotheend