Découvrez comment Ultralytics s'attaque à la pollution des océans par les plastiques en utilisant des AUV et YOLOv5 pour une détection et un nettoyage sous-marins efficaces.

Découvrez comment Ultralytics s'attaque à la pollution des océans par les plastiques en utilisant des AUV et YOLOv5 pour une détection et un nettoyage sous-marins efficaces.
Le plastique étouffe la faune marine : chaque minute, deux camions de plastique sont déversés dans nos océans, soit plus de 10 millions de tonnes par an. Les scientifiques de DeepPlastic affirment que ce plastique marin représente une menace sociétale pour "l'environnement marin, la sécurité alimentaire, la santé humaine, l'écotourisme et les contributions au changement climatique".
Pour lutter contre ce phénomène, cette équipe de chercheurs et d'ingénieurs a étudié la manière dont la vision par ordinateur peut éliminer le plastique dans nos océans.
Grâce à la technologie d'apprentissage en profondeur, les chercheurs de DeepPlastic ont mis au point une approche qui utilise des véhicules sous-marins autonomes (AUV) pour scanner, identifier et quantifier le plastique situé juste sous la surface de l'océan, là où la lumière peut encore pénétrer, c'est-à-dire dans la couche épipélagique.
"Notre objectif était de disposer d'un modèle minuscule avec une vitesse d'inférence très rapide qui puisse être utilisé pour détecter le plastique.
Jay LoweChercheur en apprentissage automatique
L'équipe de DeepPlastic a entraîné deux petits modèles précis, YOLOv4 et YOLOv5, permettant la détection d'objets en temps réel. Ces modèles ont été entraînés sur l'ensemble de données DeepTrash, qui se compose de :
Un AUV est un robot qui se déplace sous l'eau. Ce sont des véhicules lents qui peuvent glisser librement vers les profondeurs de l'océan et revenir à la surface. Un modèle d'apprentissage en profondeur doit être installé dans les AUV pour qu'ils puissent identifier et collecter le plastique sous l'eau. Les AUV peuvent être déployés en trois étapes simples pour détecter le plastique sous l'eau.
1. Installer un modèle d'apprentissage profond dans un AUV
2. Balayer l'océan
3. Identifier le plastique
L'équipe DeepPlastic a testé plusieurs modèles d'apprentissage profond tels que YOLOv4 et Faster R-CNN sur des AUV. Cependant, les chercheurs ont été confrontés à une série de défis qui ont rendu le nettoyage des océans problématique.
En l'absence d'experts en apprentissage profond au sein de l'équipe, les chercheurs n'ont pas pu tirer le meilleur parti des modèles d'apprentissage profond.
L'inférence est la vitesse à laquelle l'AUV peut reconnaître le plastique. Avec YOLOv4 et Faster R-CNN, les AUV n'ont pas été aussi efficaces dans la détection du plastique, ce qui a nui à leur capacité à nettoyer l'eau.
YOLOv4 et Faster R-CNN n'ont obtenu qu'un taux de réussite moyen de 77 % à 80 % pour l'identification du plastique.
Lors de l'utilisation du R-CNN plus rapide, 3 à 5 % des coraux ont été identifiés comme étant en plastique par les AUV, ce qui est inférieur à la norme acceptable.
En passant à YOLOv5, les chercheurs ont constaté une transformation immédiate. La précision a augmenté, la vitesse a été maximisée et la simplicité de YOLOv5 l'a rendu accessible à tous les membres de l'équipe.
Vitesse d'inférence 20% plus rapide en moyenne par rapport à un R-CNN plus rapide
Taux de précision de 93
Moins d'une heure pour installer YOLOv5
Plusieurs aspects de YOLOv5 ont permis à l'équipe de travailler facilement avec lui, en se basant sur le processus simple, étape par étape, que nous avons établi sur le référentiel.
YOLOv5 a présenté des vitesses d'inférence 20% plus rapides que Faster RCNN, traitant en moyenne une image en 9 millisecondes. Par conséquent, les AUV ont pu détecter plus rapidement le plastique flottant, ce qui a permis d'augmenter la quantité de plastique capturé et l'efficacité globale du projet.
Les taux de précision ont atteint une moyenne de 85 %, avec parfois des pointes à 93 %. Il s'agit d'un bond par rapport à la moyenne de 77 à 80 % observée avec les modèles précédents.
La mise en place de YOLOv5 a été une expérience à la fois transparente et sans effort pour les chercheurs. Les utilisateurs ont été guidés de A à Z tout au long du processus d'installation, ce qui a permis à l'équipe de commencer à utiliser YOLOv5 en moins d'une heure.
En quelques jours, à l'aide d'un petit ensemble de données de 3 000 images sans augmentation, le groupe a pu entraîner les AUV à travailler dans des lacs et des rivières. Malgré l'eau trouble et d'autres conditions défavorables, les AUV formés sur YOLOv5 ont pu détecter et identifier le plastique avec une grande précision.
"Nous recherchions un algorithme de détection d'objets qui soit à la fois très précis et extrêmement rapide. Les environnements océaniques dans lesquels nous travaillons sont des terrains difficiles et accidentés. YOLOv5 s'est avéré être le meilleur modèle de détection d'objets que nous ayons pu utiliser.
"Nous aimons utiliser YOLOv5 car il est très facile à mettre en place et à utiliser, et il a produit les résultats que nous souhaitions de manière constante.
"Pour tous les modèles que nous déploierons à l'avenir, nous considérerons YOLOv5 comme notre premier choix, sans l'ombre d'un doute.
Gautam TataChercheur en apprentissage automatique
Consultez le référentiel DeepPlastic, l'article publié et la vidéo récapitulative.