了解PyTorch,这是一个灵活的Python 机器学习框架,为Ultralytics YOLO 等人工智能创新提供动力。今天就开始更智能、更快速地构建!
PyTorch 是一个主要的开源 机器学习(ML)和 深度学习(DL)框架,可促进 智能系统的开发。PyTorch 最初由Meta AI、 现在由独立的PyTorch 基金会管理,以确保 中立和社区驱动的发展。它以其灵活性和 "Pythonic "设计而闻名,允许 开发人员构建复杂的 神经网络(NN)架构。 在Python 生态系统中感觉自然而直观。
该框架的核心是对张量进行操作。 是多维数组,类似于 NumPy库中的多维数组类似。但是 与标准数组不同,这些数据结构可以在一个 GPU来大大加快 计算速度。这种能力对于处理大规模并行处理是必不可少的。 计算机视觉 (CV) 和自然语言等任务的现代人工智能模型 计算机视觉 (CV)和自然语言 理解。
PyTorch 有别于其他框架,它采用了一系列特定的设计方案,优先考虑开发人员的 效率和调试便利性:
autograd 自动计算梯度
自动计算梯度--即
反向传播.这简化了
实现
优化算法 在培训期间。
这一框架的灵活性使其在各行各业的高影响力应用中得到广泛采用。 应用:
要了解PyTorch 在开发者工具包中的位置,最好将其与相关技术区分开来:
全部 Ultralytics YOLO11模型都是基于PyTorch 原生构建的。 这可以确保用户受益于该框架的速度和广泛的社区支持。无论是在 在自定义数据集上进行迁移学习,还是 为边缘计算部署模型,底层架构都会利用PyTorch 张量和梯度。
即将推出的Ultralytics 平台进一步简化了这一体验、 为训练和管理这些模型提供了一个精简的界面,而无需编写大量的模板代码。 代码。
下面的示例演示了如何加载预训练模型并运行推理,展示了该框架如何 如何处理繁重的计算:
from ultralytics import YOLO
# Load a standard YOLO11 model (built on PyTorch)
model = YOLO("yolo11n.pt")
# Perform object detection on an image
# PyTorch handles the tensor operations and GPU acceleration automatically
results = model("https://ultralytics.com/images/bus.jpg")
# Print the number of objects detected
print(f"Detected {len(results[0].boxes)} objects.")