- 博客(2)
- 收藏
- 关注
RSS订阅原创 yolov5改mobilevit遇到的问题
虽然确实轻量化了，但代价太大了，掉点严重，且非常吃显存，batch-size为4，workers为1，显存占了8.35G，一个epoch的时间是原版yolov5s的3倍，感觉这个改进很不划算，可以作为背景板放进对比实验中。另外，改backbone其实本身逻辑就有问题，backbone作为网络的主干，把它换了，那还和它有关系吗？类似于改装自行车，直接把车架换了。所以，轻量化不该只考虑空间复杂度（参数量），时间复杂度（计算量），模型体积，也得考虑同等占显存情况下的训练的速度。
2023-07-26 23:49:22
285
3
原创 YOLOv5s配置环境时遇到的问题
3.预训练权重下载方法：进入gitcode后，点击页面右侧的“全部发行版”，然后找到yolov5s.pt，之所以要下载它，是因为在代码中，train.py文件中444行weights参数，默认就是yolov5s.pt，如果下载了其他的预训练权重，记得代码中要修改，还有就是权重放在根目录下。1.在detect之前，应该要先pip install -r requirements.txt，当然这也不是什么问题，但像我这样的萌新确实有可能在遇到问题后，一个一个去pip。
2023-07-15 15:40:11
355
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人