Glossaire

PyTorch

Découvre PyTorch, le cadre flexible d'apprentissage profond pour construire, former et déployer des réseaux neuronaux avec l'accélération GPU .

Entraîne les modèles YOLO simplement
avec Ultralytics HUB

En savoir plus

PyTorch est un cadre d'apprentissage automatique open-source qui accélère le chemin entre le prototypage de la recherche et le déploiement de la production. Développé principalement par le laboratoire de recherche en IA de Facebook (FAIR), il est devenu une pierre angulaire dans le domaine de l'apprentissage profond en raison de sa flexibilité, de sa facilité d'utilisation et de sa forte prise en charge de l'accélération GPU . PyTorch est largement utilisé à la fois dans le milieu universitaire et dans l'industrie pour une variété d'applications, y compris la vision par ordinateur (CV), le traitement du langage naturel (NLP), et bien plus encore. Pour les utilisateurs familiers avec les concepts de base de l'apprentissage automatique, PyTorch offre une plateforme intuitive pour construire, former et déployer des modèles sophistiqués.

Principales caractéristiques et avantages

PyTorch est connu pour son graphe de calcul dynamique, qui permet une plus grande flexibilité dans la construction de modèles complexes par rapport aux graphes statiques utilisés dans d'autres frameworks comme . TensorFlow. Cette nature dynamique signifie que le graphe est défini à la volée, ce qui facilite le débogage et permet un développement de modèle plus intuitif. De plus, l'intégration transparente de PyTorch avec Python et son vaste écosystème de bibliothèques et d'outils améliorent encore sa facilité d'utilisation. Sa forte prise en charge de l'accélération GPU (Graphics Processing Unit) permet d'accélérer la formation des modèles d'apprentissage profond (DL), ce qui est particulièrement bénéfique pour les grands ensembles de données et les réseaux neuronaux (NN) complexes.

Composantes essentielles

PyTorch est construit autour du concept de tenseurs, qui sont des tableaux multidimensionnels similaires aux ndarrays de NumPy, mais avec la capacité supplémentaire de fonctionner sur les GPU. Le cadre fournit un ensemble complet d'outils pour les opérations tensor , la différenciation automatique et la construction de réseaux neuronaux. Les torch.nn offre une API de haut niveau pour la construction et l'entraînement des réseaux neuronaux, tandis que le module torch.optim Le module fournit divers algorithmes d'optimisation comme Descente stochastique de gradient (SGD) et Adam Optimizer. AutogradPyTorch's automatic differentiation package, calcule automatiquement les gradients, ce qui est crucial pour l'entraînement des réseaux neuronaux à l'aide de rétropropagation.

Applications dans le monde réel

PyTorch est utilisé dans un large éventail d'applications réelles dans diverses industries. Voici deux exemples concrets :

  1. Santé: Dans le secteur de la santé, PyTorch est utilisé pour l'analyse d'images médicales, comme la segmentation de tumeurs dans les IRM ou la détection d'anomalies dans les radiographies. Par exemple, les chercheurs utilisent les réseaux de neurones convolutifs (CNN) sur PyTorch pour construire des modèles capables d'identifier et de classer avec précision les maladies à partir d'images médicales, ce qui facilite les diagnostics précoces et la planification des traitements. En savoir plus sur l'IA dans le domaine de la santé.
  2. Véhicules autonomes: PyTorch joue un rôle crucial dans le développement des voitures auto-conduites. Il est utilisé pour former des modèles d'apprentissage profond pour des tâches telles que la détection d'objets, la détection de voies et la planification de trajectoires. Par exemple, les modèles Ultralytics YOLO (You Only Look Once), qui sont souvent mis en œuvre à l'aide de PyTorch, permettent de détecter des objets en temps réel dans les flux vidéo des caméras des véhicules, ce qui permet à la voiture de percevoir son environnement et d'y réagir. En savoir plus sur l'IA dans les voitures auto-conduites.

PyTorch vs. TensorFlow

Bien que PyTorch et TensorFlow soient des cadres d'apprentissage profond populaires, ils présentent des caractéristiques distinctes. TensorFlow Le premier, développé par Google, utilise un graphe de calcul statique, ce qui signifie que l'ensemble de la structure du graphe doit être défini avant d'exécuter le modèle. Cela peut conduire à une exécution plus rapide mais à moins de flexibilité par rapport au graphe dynamique de PyTorch. TensorFlow dispose d'un écosystème de déploiement plus important, avec des outils tels que TensorFlow Serving et TensorFlow Lite pour déployer des modèles en production et sur des appareils mobiles. PyTorch En revanche, le graphique dynamique de est souvent privilégié pour la recherche en raison de sa flexibilité et de sa facilité d'utilisation, en particulier dans les milieux universitaires.

Commencer avec PyTorch

Pour commencer à travailler avec PyTorch, tu peux explorer le site officiel PyTorch , qui fournit une documentation complète, des tutoriels et une assistance à la communauté. En outre, la documentation deUltralytics offre des ressources précieuses pour l'utilisation de PyTorch avec les modèles Ultralytics YOLO , y compris des guides sur la formation, la validation et le déploiement des modèles. Tu peux également explorer des modèles pré-entraînés et commencer à entraîner tes propres modèles en utilisant le Ultralytics HUB, qui simplifie le processus de construction et de déploiement des modèles d'IA de la vision.

Tout lire