Ultralytics_yolov10目标检测,预处理函数入口

日期:2024.10.7.

随着Ultralytics的更新,yolov5-v11可以统一使用Ultralytics包体,我之前分析的yolov5关键代码定位在Ultralytics中不适用,这篇博客更新一下。

1. Ultralytics包体版本:

bash 复制代码
$ pip list | grep ultralytics
ultralytics                  8.2.101
ultralytics-thop             2.0.8

2. 测试代码

我不确定yolov10的预处理入口和其他yolo版本是否一样,大家感兴趣自己测试一下

python 复制代码
from ultralytics import YOLO

# Load a model
model = YOLO("yolov10n.pt")  # build a new model from YAML

# Train the model with data augmentation enabled
results = model.train(
    data="your_path1/your_dataset.yaml",
    epochs=150,
    imgsz=320,
    batch=8,
    project='your_path2',
    name='train',
    pretrained=True,
)

3. train阶段预处理入口

  1. 文件:ultralytics/data/base.py
  2. 类:class BaseDataset(Dataset):
  3. 函数:def load_image(self, i, rect_mode=True):

在base.py文件中的---->BaseDataset类---->load_image函数

4. val阶段预处理函数入口

  1. 文件:ultralytics/data/augment.py
  2. 类:class LetterBox:
  3. 函数:def call(self, labels=None, image=None):

在augment.py文件中的---->LetterBox类---->__call__函数

5. 验证方法

5.1. 方法1:

修改预处理方法,查看训练结果存放的文件夹your_path2/train,这个文件夹在第2节的测试代码中指定,文件夹内可以看到train和val阶段的图片

5.2. 方法2

打断点查看进入WarmupForward的图片,打断点的位置:

  1. 文件:ultralytics/engine/trainer.py
  2. 类:class BaseTrainer:
  3. 函数:def _do_train(self, world_size=1):
  4. 在函数中找到for i, batch in pbar:并打断点

对batch中的img,查看或保存到本地就可以验证了。

python 复制代码
images = batch['img']
save_path = "./image_{}.jpg"  # 用你自己的路径

# 遍历每一张图片
for i in range(images.shape[0]):
	# 获取第 i 张图片的 Tensor,形状 [3, 320, 320]
	img_tensor = images[i]
	
	# 调整维度顺序,从 [3, 320, 320] -> [320, 320, 3]
	img_np = img_tensor.permute(1, 2, 0).numpy()
	
	# 将Tensor数值范围从[0, 1](假设这是你的数据范围)转换为[0, 255],并转换为uint8类型
	img_np = (img_np).astype(np.uint8)
	
	# 使用 PIL Image 将 NumPy 数组转换为图像
	img_pil = Image.fromarray(img_np)
	
	# 保存图片
	img_pil.save(save_path.format(i))  # 保存图片路径中带有索引i
相关推荐
Michelle8023几秒前
23大数据 数据挖掘复习1
大数据·人工智能·数据挖掘
ayaya_mana6 分钟前
Linux环境下Node.js任意版本安装与pnpm、yarn包管理
linux·node.js·vim
IT学长编程8 分钟前
计算机毕业设计 基于k-means的校园美食推荐系统 Python 大数据毕业设计 Hadoop毕业设计选题【附源码+文档报告+安装调试】
大数据·python·毕业设计·kmeans·课程设计·毕业论文·美食推荐系统
乌龙玛奇朵5199 分钟前
Finalshell建立连接
linux
如竟没有火炬35 分钟前
LRU缓存——双向链表+哈希表
数据结构·python·算法·leetcode·链表·缓存
Serverless 社区42 分钟前
阿里云函数计算 AgentRun 全新发布,构筑智能体时代的基础设施
人工智能·阿里云·云原生·serverless·云计算
weixin_4569042744 分钟前
YOLOv11安卓目标检测App完整开发指南
android·yolo·目标检测
Maple_land1 小时前
Linux进程第八讲——进程状态全景解析(二):从阻塞到消亡的完整生命周期
linux·运维·服务器·c++·centos
嵌入式分享1 小时前
嵌入式分享#41:RK3576改UART波特率【精简版】
linux·嵌入式硬件·ubuntu·嵌入式