YOLOv5来了
软硬件环境 ubuntu 18.04 64bit anaconda with 3.7 nvidia gtx 1070Ti cuda 10.1 pytorch 1.5 YOLOv5 前言 YOLOv4 还没有退热, YOLOv5 就已经来了! 6月9日, Ultralytics 公司开源了 YOLOv5 ,离上一次 YOLOv4 发布不到50天,不过这一次的 YOLOv5 是基于 PyTorch 实现的,而不是之前版本的 darknet ! 根据官方给出的数字, YOLOv5 的速度最快可以达到每秒140帧( FPS ),但是权重文件只有 YOLOv4 的1/9,而且准确度更高。本次的发布的 YOLOv5 并不是一个单独的模型,而是一个模型家族,包括了 YOLOv5s 、 YOLOv5m 、 YOLOv5l 、 YOLOv5x ,要求 Python 3.7和 PyTorch 1.5以上版本。 关于 YOLOv5 这个版本,大家可以看看知乎中的讨论,链接放在文末的参考资料中 安装GPU环境 请参考之前的文章 ubuntu安装cuda windows 10安装cuda和cudnn 安装pytorch的GPU版本 来到官网 https://pytorch.org/get-started/locally/,根据自己的环境,进行选择,网站会给出相应的安装命令。我这里的环境是 linux 、