yolov5复现pytorch教程
时间: 2023-10-06 11:04:12 浏览: 157
yolo论文与pytorch复现版
5星 · 资源好评率100%
在学习目标检测时,我找到了一个成功运行的YOLOv1 pytorch复现版本,这个版本是由一位博主在GitHub上上传的资源实现的。我建议你直接去看这位博主的版本,可以下载并用pycharm加载出来。然后,你可以对照我所给出的主要模块的注释(源代码注释很少,我基本都注释了),就可以很轻松地复现出来。这个复现版本中,模型的前半部分卷积网络采用了resnet50,并在官方训练好的resnet50网络参数上进行了训练。请注意,由于模型较大,可能需要一些时间来下载和加载。
阅读全文