保姆级教程:在Ubuntu22.04上5分钟搞定YOLOv8的安装与五大任务初体验(附CUDA11.7+Pytorch1.13配置)
5分钟极速部署YOLOv8Ubuntu 22.04环境下的全功能实战指南刚拿到新装的Ubuntu系统与RTX显卡时最令人兴奋的莫过于快速验证深度学习框架的实战能力。YOLOv8作为当前目标检测领域最受欢迎的算法之一其开箱即用的特性尤其适合快速验证。本文将带你用最简洁的步骤在纯净的Ubuntu 22.04系统中完成从环境配置到五大核心功能的完整体验。1. 环境准备与一键配置在开始前请确保系统已安装NVIDIA驱动推荐使用470以上版本。打开终端依次执行以下命令创建隔离的Python环境# 创建Python 3.8虚拟环境YOLOv8对3.8支持最稳定 conda create -n yolov8_env python3.8 -y conda activate yolov8_env接下来安装匹配CUDA 11.7的PyTorch和Ultralytics包# 安装PyTorch 1.13 CUDA 11.7 pip install torch1.13.1cu117 torchvision0.14.1cu117 --extra-index-url https://download.pytorch.org/whl/cu117 # 安装YOLOv8核心库 pip install ultralytics注意如果遇到权限问题可添加--user参数或使用pip install --upgrade pip确保pip版本最新验证安装是否成功python -c from ultralytics import YOLO; print(YOLO(yolov8n.pt))2. 五大核心功能实战演练2.1 目标检测Detection新建工作目录并执行检测命令mkdir yolo_test cd yolo_test yolo taskdetect modepredict modelyolov8n.pt sourcehttps://ultralytics.com/images/bus.jpg执行后会在runs/detect/predict目录生成带检测框的结果图像。常用参数conf: 置信度阈值默认0.25imgsz: 输入图像尺寸默认640save_txt: 保存检测结果为TXT文件2.2 实例分割Segmentation对视频进行实时分割处理yolo tasksegment modepredict modelyolov8n-seg.pt sourceyour_video.mp4分割任务会额外生成掩码图像PNG格式多边形坐标文件JSON格式2.3 图像分类Classification使用预训练分类模型yolo taskclassify modepredict modelyolov8n-cls.pt sourceimage.jpg输出结果包含Top-5类别预测各类别置信度分数2.4 姿态估计Pose Estimation人体关键点检测示例yolo taskpose modepredict modelyolov8n-pose.pt source0 # 使用摄像头输出包含17个COCO格式的关键点坐标可视化效果如下关键点ID对应部位0鼻子1-2左右眼3-4左右耳5-6左右肩2.5 模型训练与验证使用COCO128数据集快速验证训练流程# 训练自动下载数据集 yolo train modelyolov8n.pt datacoco128.yaml epochs10 imgsz640 # 验证 yolo val modelpath/to/best.pt datacoco128.yaml训练过程会实时显示指标变化Epoch gpu_mem box obj cls labels img_size 1/10 1.2G 0.123 0.456 0.789 32 640 2/10 1.2G 0.098 0.321 0.654 29 6403. 高级技巧与性能优化3.1 模型导出与部署将PyTorch模型转换为ONNX格式yolo export modelyolov8n.pt formatonnx支持导出格式包括TensorRT.engineCoreML.mlmodelOpenVINO.xml3.2 多GPU训练加速使用DDP模式进行分布式训练python -m torch.distributed.run --nproc_per_node 2 yolo train modelyolov8n.pt datacoco128.yaml3.3 自定义数据集训练准备数据集目录结构custom_data/ ├── images/ │ ├── train/ │ └── val/ └── labels/ ├── train/ └── val/创建数据集配置文件custom.yamlpath: ./custom_data train: images/train val: images/val names: 0: class1 1: class24. 常见问题解决方案Q1CUDA out of memory错误降低batch-size默认16减小imgsz尺寸使用--device 0指定单GPUQ2预测结果不准确尝试更大的模型如yolov8x.pt调整conf参数过滤低置信度结果检查输入图像是否经过正常预处理Q3视频预测卡顿添加--half使用FP16推理设置streamTrue启用流式处理降低视频分辨率后再处理在RTX 3060显卡上的基准测试表现模型推理速度msmAP50YOLOv8n12.30.675YOLOv8s15.80.712YOLOv8m25.60.753
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2559305.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!