深圳Yolo 视觉
深圳
立即加入
词汇表

PyTorch

了解PyTorch,这是一个灵活的Python 机器学习框架,为Ultralytics YOLO 等人工智能创新提供动力。今天就开始更智能、更快速地构建!

PyTorch ch是由Meta AI主导开发的开源机器学习库PyTorch 深度学习领域研究者和开发者的基石PyTorch 凭借其灵活性和易用性,该库允许用户通过动态计算图构建和训练复杂神经网络。 该特性常被称为"即时执行",能使代码立即评估, 相较于依赖静态图定义的框架, 其调试和原型开发过程更为直观。 它与Python 语言无缝集成, 宛如标准科学计算工具的自然延伸。

核心机制与重要性

该框架的核心是张量,其本质是多维数组,NumPy 中描述的数组。但与标准数组不同PyTorch 专为NVIDIA CUDA提供的GPU 而设计。这种硬件加速对于高效训练现代人工智能(AI)模型所需的大规模并行处理至关重要。

该库支持一个庞大的工具生态系统,涵盖计算机视觉(CV)和自然语言处理领域。通过提供丰富的预构建层、优化器和损失函数,它简化了图像分类和序列建模等任务的算法创建过程。

实际应用

该框架的多功能性使其被广泛应用于不同行业,用于开发高影响力的AI解决方案:

  1. 自动驾驶汽车 行业领军企业PyTorch 深度学习模型PyTorch 处理来自车载摄像头的视频流。这些模型通过实时物体检测识别车道、路标及行人,从而实现更安全的导航。
  2. 医疗诊断:研究人员利用该框架开发先进的医疗应用。例如,它驱动着分析MRI扫描或X光片的系统,通过精确的图像分割技术协助医生检测肿瘤。

与相关工具的比较

为更好地理解其作用,有必要PyTorch 人工智能技术栈中的其他常用工具区分开来:

  • TensorFlow对比:由Google开发的TensorFlow 依赖静态计算图,这使得调试难度增加但优化了部署。尽管两者功能已趋近PyTorch 因其直观的界面,在快速原型开发和研究PyTorch 优先选用。
  • OpenCV对比: OpenCV 传统图像处理功能(如尺寸调整、滤波和色彩转换)OpenCV 库,而非训练神经网络在典型工作流中,开发者会先使用OpenCV 数据预处理,再将图像PyTorch 进行分析。

与Ultralytics集成

Ultralytics ytics Ultralytics 家族,包括尖端的 YOLO26和广泛使用的 YOLO11,均原生构建PyTorch。这一基础架构 确保用户能充分受益于该框架的速度、稳定性及广泛的社区支持。无论是针对定制训练数据执行 迁移学习,还是为边缘设备导出模型,其底层架构 始终依托PyTorch 梯度计算。

即将Ultralytics 将进一步简化这一体验,提供统一界面来管理数据集获取、训练和部署,无需编写大量冗余代码。

以下示例演示了如何验证GPU 并使用YOLO 进行推理,展示了该框架如何在后台处理硬件加速:

import torch
from ultralytics import YOLO

# Check if CUDA (GPU) is available for PyTorch acceleration
device = "cuda" if torch.cuda.is_available() else "cpu"
print(f"PyTorch is using device: {device}")

# Load a YOLO26n model (built on PyTorch)
model = YOLO("yolo26n.pt")

# Perform object detection on an image
# PyTorch handles tensor operations and moves data to the GPU automatically
results = model("https://ultralytics.com/images/bus.jpg", device=device)

加入Ultralytics 社区

加入人工智能的未来。与全球创新者联系、协作和共同成长

立即加入