研究了这么久的yolo姿态算法终于入门啦!!!!
那么接下来由我带领大家进入yolo世界,首先安装软件,需要vscode,python,pycharm以及Anaconda(它的下载路径不能有中文)。具体安装方法搜一下就有了,本文不详细介绍喽。还需要到网站去下载开源代码,当然你也可以进我主页找到对应资源包去下载。
代码网址:GitHub - ultralytics/ultralytics: NEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLite
代码和软件安装好了之后,就可以配置环境了。打开Anaconda powershell prompt,进来之后,直接输入命令:conda create -n ultralytics python=3.9,成功后会出现下图
之后激活ultralytics的环境,输入命令:conda activate ultralytics
之后,我们就开始在这下载安装包,使用清华源镜像,输入命令:pip config set global.index-url https://pypi.tuna.tsinghua.tsinghua.edu.cn/simple
当然,你也可以使用其他镜像源如下,也可以不使用,但是下载速度估计会变慢,这里推荐镜像一下。
清华大学开源软件镜像站:https://pypi.tuna.tsinghua.edu.cn/simple
阿里云开源镜像站:https://mirrors.aliyun.com/pypi/simple/
豆瓣:https://pypi.douban.com/simple/
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
继续安装pytorch,先去官网:https://pytorch.org/,选择安装的版本,有英伟达的显卡就选择CUDA的版本,你安装的版本低于现在英伟达显卡的版本就行。如果没有就是安装CPU的,选择好直接复制命令就行(Run This Command 那栏)。还是在Anaconda这个界面输入。
接下来进入到你从官网下载的开源代码所在的位置,我以我的位置为例:
进入之后输入安装命令:pip install -e .
安装好之后,再安装 pip install jupyterlab tensorboard
最后检查一下是否安装好了 输入命令yolo
接下来进入vscode,首先要自己激活vscode的python部分(教程一抓一大把,找一个配置就可以),在商城安装插件,包括:python(代码编写使用),jupyter(编写可用),remotessh(连接服务器),Chinese Language Pack(简体中文包),Opencv(提供资源包)
在vscode打开你在官网下载的ultralytics,然后配置终端,点击新建终端,
点击之后,会出默认的base终端,换成我们配置好的就行,点击右下角
选择带这个的ultralytics环境,把这个终端删了,再起一个新的就可以进行编写了。新建一个,my train.py 的文件。输入代码:
from ultralytics import YOLO
# Load a model
model = YOLO("yolov8n.pt") # load an official model
# Predict with the model
results = model(source="./ultralytics/assets/bus.jpg",save=True) # predict on an image
选中代码,右键,选择在终端中运行python。
接下来等待,再从runs下的detect下找到识别好的图片,就可以看到结果啦。我使用的是yolov8n.pt的训练权重。
内容好多,写不完了,关于后续的模型训练以及姿态识别,等待下期,后面识别不出来,不要慌,看我别的博文,可以解决。over,下篇见。
🎈如果本篇博文对你有益,可以打赏一点,你的鼓励将是我最大的动力🌹。
(ps:一分两分就可以🌷)