天天看点

安装测试YOLOv5

不久前,刚刚发布了YOLOv5版本的源代码,并且可以在github网站下下载,笔者今天下载下来试用了一下,对于遇到的一些问题做个记录,希望能让大家少走弯路。

YOLOv5网址:https://github.com/ultralytics/yolov5

访问该网址,在Ubuntu16.04的系统中进行安装:

pip install -U -r requirements.txt
           

能够顺利安装完成。随后按照提示,运行如下命令进行测试:

$ python detect.py --source file.jpg
           

确发现报错了,具体提示信息为:

TypeError: _resolve_type_from_object(): incompatible function arguments. The following argument types are supported: 
1. (arg0: object, arg1: torch._C._jit_tree_views.SourceRange, arg2: Callable[[str], function]) -> torch._C.Type
           

更多信息如图所示:

安装测试YOLOv5

这到底是什么原因呢?原来YOLOv5的网址中已经注明了需要Python的版本中3.7以上,而从上图可以看出,Python的版本是3.5,所以出错了。升级或者重新安装3.7以上版本的Python,然后运行命令:

python3.7 detect.py --source  image.png
           

最终显示结果如下:

安装测试YOLOv5

识别图像的结果如下:

安装测试YOLOv5

我们可以看到,一辆车,四位交警同志,都识别出来了。效果还是不错的。注意第一次运行python detect.py的时候,可能会运行较长时间才能出结果,因为需要从网络中下载一些数据才行。

------------------------------ 正文到此结束---------------------------------

推荐一个公众号:健哥聊量化,会持续推出股票相关基础知识,以及python实现的一些基本的分析代码。欢迎大家关注,二维码如下:

安装测试YOLOv5

相关文章列表如下:

  • 技术炒股VS价值投资
  • 股票基础知识----- K线形态
  • 股票K线形态 ----早晨之星
  • “早晨之星”实际操作篇---通达信软件为例
  • 牛刀小试----python+tushare进行股票分析
  • ​股票K线形态----黄昏之星