![](https://img-blog.csdnimg.cn/20201014180756757.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
YOLOv5
文章平均质量分 71
晴明大大
这个作者很懒,什么都没留下…
展开
-
【Pytorch】修改网络后加载预训练权重
内容本文章带大家如何给自己修改过后的网络,加载预训练权重。很多小伙伴针对某一模型进行修改的时候,在修改模型后想要加载预训练权重,会发现频频报错,其实最主要原因就是权重的shape对应不上。注意:以下方法仅仅针对于在原网络改动不大的情况下加载预训练权重!1、.pt文件----->model:从.pt文件直接加载预训练权重。# 模板ckpt = torch.load(weights) # 加载预训练权重model = Model() # 创建我们的模型model_dict = mode原创 2022-05-16 19:07:21 · 11917 阅读 · 17 评论 -
YOLOv5 + Win10 训练DJI官方机器人数据集【1】YOLOv5环境搭建
目录一、环境搭建1.1安装列表1.2 安装1.2.1 Anaconda安装1.2.2 CUDA10.2 安装1.2.3 Pytorch安装1.2.4 Opencv安装1.2.5 YOLOv5 安装2、测试yolov5二、可能遇见的问题初试完,回到实验室再次接触到本科做过的RM比赛,想玩一玩yolov5,搞了一晚上总算能训练了(虽然原理还不太明白qwq),中间遇到了很多坑。由于懒得用ubuntu了,就直接搞了win10版的。下面就是我配置的过程,给大家排排坑。一原创 2021-04-06 11:27:28 · 1490 阅读 · 1 评论