Yolo Vision Shenzhen
Shenzhen
Rejoindre maintenant
Glossaire

PyTorch

Découvrez PyTorch, le framework d'apprentissage machine flexible, Python, qui alimente des innovations en IA comme Ultralytics YOLO. Construisez plus vite et plus intelligemment dès aujourd'hui !

PyTorch is an open-source machine learning library primarily developed by Meta AI that has become a cornerstone for researchers and developers in the field of deep learning. Known for its flexibility and ease of use, it allows users to build and train complex neural networks using a dynamic computational graph. This feature, often referred to as "eager execution," enables code to be evaluated immediately, making debugging and prototyping significantly more intuitive compared to frameworks that rely on static graph definitions. It integrates seamlessly with the Python programming language, feeling like a natural extension of standard scientific computing tools.

Core Mechanics and Significance

At the heart of this framework are tensors, which are multi-dimensional arrays similar to those found in the NumPy documentation. However, unlike standard arrays, PyTorch tensors are designed to leverage GPU acceleration provided by NVIDIA CUDA. This hardware acceleration is critical for the massive parallel processing required to train modern artificial intelligence (AI) models efficiently.

The library supports a vast ecosystem of tools for computer vision (CV) and natural language processing. By providing a rich set of pre-built layers, optimizers, and loss functions, it simplifies the process of creating algorithms for tasks like image classification and sequence modeling.

Applications concrètes

La polyvalence de ce cadre a conduit à son adoption dans divers secteurs pour des solutions d'IA à fort impact :

  1. Autonomous Vehicles: Industry leaders leverage PyTorch to build deep learning models that process video feeds from car cameras. These models perform real-time object detection to identify lanes, signs, and pedestrians, enabling safer navigation.
  2. Medical Diagnostics: Researchers use the framework to develop advanced healthcare applications. For instance, it powers systems that analyze MRI scans or X-rays to assist doctors in detecting tumors via precise image segmentation.

Comparaison avec des outils similaires

Pour mieux comprendre son rôle, il est utile de distinguer PyTorch autres outils courants de la pile IA :

  • Vs. TensorFlow: développé par Google, TensorFlow s'appuyait TensorFlow sur des graphes computationnels statiques, ce qui rendait le débogage plus difficile mais optimisait le déploiement. Bien que les deux frameworks aient convergé en termes de fonctionnalités, PyTorch souvent préféré pour le prototypage rapide et la recherche en raison de son interface intuitive.
  • Vs. OpenCV: OpenCV une bibliothèque axée sur les fonctions traditionnelles de traitement d'images (telles que le redimensionnement, le filtrage et la conversion des couleurs) plutôt que sur l'entraînement des réseaux neuronaux. Dans un flux de travail type, les développeurs utilisent OpenCV le prétraitement des données avant d'introduire les images dans un PyTorch pour analyse.

Intégration avec Ultralytics

Toute la gamme Ultralytics , y compris le modèle de pointe YOLO26 et le très répandu YOLO11, est construite de manière native sur PyTorch. Cette base garantit aux utilisateurs de bénéficier de la vitesse, de la stabilité et du soutien étendu de la communauté du framework. Qu'il s'agisse d'effectuer un transfert d'apprentissage sur des données d'entraînement personnalisées ou d'exporter des modèles pour des appareils de pointe, l' architecture sous-jacente s'appuie sur PyTorch et les gradients PyTorch .

La future Ultralytics simplifie encore davantage cette expérience, en fournissant une interface unifiée pour gérer l'approvisionnement, la formation et le déploiement des ensembles de données sans avoir à écrire de code standardisé.

L'exemple suivant montre comment vérifier GPU et exécuter une inférence à l'aide d'un YOLO , illustrant ainsi comment le framework gère l'accélération matérielle en arrière-plan :

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)

Rejoindre la communauté Ultralytics

Rejoignez le futur de l'IA. Connectez-vous, collaborez et évoluez avec des innovateurs mondiaux.

Rejoindre maintenant