YOLO-Master 与 YOLO26 开始

2026 年初,YOLO 迎来了新变化,都非常亮眼 👍

一是 YOLO-Master:引入了 ES-MoE,让计算更智能。会依据输入自适应做计算,不再是静态的。

二是 YOLO26:更简化的设计,直接端到端出检测结果。不用再搞后处理了呢,利于边缘做部署。

既然 YOLO 上新了,那一起玩一玩吧 ☺️

以下则是动手实践了呢。

YOLO-Master

环境

准备 Conda 环境,

bash 复制代码
conda create -n yolom python=3.12
conda activate yolom

# Install PyTorch (CPU version)
pip install torch torchvision
# Install PyTorch with CUDA (version <= nvidia-smi shown)
#  https://pytorch.org/get-started/locally
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu130

准备 YOLO-Master,

bash 复制代码
# Clone the repository
git clone --depth 1 https://github.com/Tencent/YOLO-Master
cd YOLO-Master

# Install dependencies
pip install -r requirements.txt
pip install -e .

# Optional: Install FlashAttention for faster training (CUDA required)
#   https://github.com/Dao-AILab/flash-attention
pip install flash_attn

准备模型,

bash 复制代码
# 获取模型,YOLO-Master-EsMoE-N
wget https://huggingface.co/gatilin/YOLO-Master-ckpts-v0/resolve/main/YOLO-Master-EsMoE-N/YOLO-Master-EsMoE-N.pt?download=true
ln -s YOLO-Master-EsMoE-N.pt yolo_master_n.pt

推理

python 复制代码
from ultralytics import YOLO

# 加载模型,Nano 版
model = YOLO("yolo_master_n.pt")
# model = YOLO("runs/detect/train/weights/best.pt")

# 检测图像
results = model.predict("data/dog.jpg")
results[0].show()

# 保存结果
results[0].save("result/dog.jpg")

训练

python 复制代码
from ultralytics import YOLO

# 加载模型,从 YAML 构建(从零开始)
# model = YOLO('cfg/models/master/v0/det/yolo-master-n.yaml')
# 加载模型,Nano 版(全量微调)
model = YOLO("yolo_master_n.pt")

# 训练模型,用 COCO8 数据集测试
#   结果在 runs/detect/train10/weights/best.pt
results = model.train(
    data='coco8.yaml',  # data='coco.yaml',
    epochs=100,         # epochs=600,
    batch=8,            # batch=256,
    imgsz=640,
    device="0",         # device="0,1,2,3",  # 如果使用多 GPU
    scale=0.5,
    mosaic=1.0,
    mixup=0.0,
    copy_paste=0.1
)

YOLO-Master 还可以 LoRA 微调 ☺️

YOLO26

环境

准备 Conda 环境,

bash 复制代码
conda create -n yolo26 python=3.12
conda activate yolo26

# Install PyTorch (CPU version)
pip install torch torchvision
# Install PyTorch with CUDA (version <= nvidia-smi shown)
#  https://pytorch.org/get-started/locally
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu130

准备 YOLO26,

bash 复制代码
# Install ultralytics package
pip install -U ultralytics

# Install dependencies
pip install faster-coco-eval

推理

python 复制代码
from ultralytics import YOLO

# 加载模型,Nano 版
model = YOLO("yolo26n.pt")
# model = YOLO("runs/detect/train/weights/best.pt")

# 检测图像
results = model.predict("data/dog.jpg")
results[0].show()

# 保存结果
results[0].save("result/dog.jpg")

训练

python 复制代码
from ultralytics import YOLO

# 加载模型,Nano 版
model = YOLO("yolo26n.pt")

# 训练模型,用 COCO8 数据集测试
#   结果在 runs/detect/train/weights/best.pt
results = model.train(data="coco8.yaml", epochs=100, imgsz=640)

结语

YOLO 训练越来越简便了呢,部署仍旧看平台咯。

相关推荐
科技大视界1 小时前
投资AI项目,传统尽调不够用了——李章虎律师拆解算法、数据、算力三大雷区
人工智能·算法·数据挖掘
郝学胜-神的一滴1 小时前
算法实战:最小k个数——大顶堆的优雅解法
开发语言·数据结构·c++·python·程序人生·算法·排序算法
Irissgwe1 小时前
算法滑动窗口
数据结构·算法
怪兽学LLM1 小时前
LeetCode 105. 从前序与中序遍历序列构造二叉树:分治递归思路详解
算法·leetcode·职场和发展
可编程芯片开发2 小时前
基于PI控制的三相整流器控制系统的simulink建模与仿真,包含超级电容充电和电机
算法
酿情师2 小时前
区块链共识算法深度拆解:PoW、PoS、PBFT、Raft 原理解析
算法·区块链·共识算法
hansang_IR2 小时前
【记录】「SCOI2016」三道模拟赛/26.7.12
c++·算法
退休倒计时2 小时前
【每日一题】LeetCode 39. 组合总和 TypeScript
算法·leetcode·typescript
不是az2 小时前
力控相关知识点
算法·机器学习·最小二乘法
战血石LoveYY3 小时前
Integer类型超限,导致数据异常
java·算法