search

bubbliiiing / yolov4-tiny-pytorch

这是一个YoloV4-tiny-pytorch的源码，可以用于训练自己的模型。
MIT License
786 stars 184 forks source link

大佬，为什么tiny的网络结构比较精简，但是感觉预训练权重文件还是比较大呀？ #81

Open LiHai-26 opened 2 years ago

LiHai-26 commented 2 years ago

因为看ultralytics的yolov5s的预训练权重只有十几M，但是网络结构肯定比yolov4-tiny复杂，yolov4-tiny就只有三个Resblock，参数量也比yolov5少，感觉权重文件应该比较小，但为什么预训练权重文件却有二十多M呢？是因为模型保存格式不一样吗？yolov5用的pt，大佬这边都用的pth，因为我看大佬实现的yolov5-pytorch版本里，yolov5s的pth预训练权重也有二十多M，不知道是不是这个原因； 而且好多轻量化模型的权重文件也只有几M，大佬有空的时候可以帮忙解答一下吗？ （因为想在大佬的yolov4-tiny基础上做实验，与yolov5做对比的时候想在模型大小和参数量上想压过yolov5s一头，但是我却是个半吊子，也不会分析模型，不知道该怎么说明）

bubbliiiing commented 2 years ago

他是half后保存的，我是全精度保存的

LiHai-26 commented 2 years ago

哦哦，原来是这样呀，谢谢大佬回复，谢谢大佬复现的项目，感谢感谢，辛苦辛苦

---原始邮件--- 发件人: @.> 发送时间: 2022年4月30日(周六) 晚上11:16 收件人: @.>; 抄送: "Hai @.**@.>; 主题: Re: [bubbliiiing/yolov4-tiny-pytorch] 大佬，为什么tiny的网络结构比较精简，但是感觉预训练权重文件还是比较大呀？ (Issue #81)

他是half后保存的，我是全精度保存的

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>