U2-Net : Approfondir la détection d'objets saillants grâce à une structure en U imbriquée

Résumé

Dans cet article, nous concevons une architecture de réseau profond simple mais puissante, U2-Net, pour la détection d'objets saillants (SOD). L'architecture de notre U2-Net est une structure en U imbriquée à deux niveaux. Cette conception présente les avantages suivants : (1) elle est capable de capturer plus d'informations contextuelles de différentes échelles grâce au mélange de champs réceptifs de différentes tailles dans nos blocs U ReSidual (RSU) proposés, (2) elle augmente la profondeur de l'ensemble de l'architecture sans augmenter significativement le coût de calcul en raison des opérations de pooling utilisées dans ces blocs RSU. Cette architecture nous permet d'entraîner un réseau profond à partir de zéro sans utiliser de backbones provenant de tâches de classification d'images. Nous instancions deux modèles de l'architecture proposée, U2-Net (176,3 Mo, 30 FPS sur GPU GTX 1080Ti) et U2-Net† (4,7 Mo, 40 FPS), pour faciliter l'utilisation dans différents environnements. Les deux modèles atteignent des performances compétitives sur six jeux de données SOD. Le code est disponible : https://github.com/NathanUA/U-2-Net.