HyperAIHyperAI

Command Palette

Search for a command to run...

il y a 4 mois

Entraînement modernisé de U-Net pour la segmentation sémantique aérienne

{Ivan Gruber Jakub Straka}

Entraînement modernisé de U-Net pour la segmentation sémantique aérienne

Résumé

Dans cet article, nous proposons un protocole d'entraînement amélioré pour l'architecture U-Net, destiné à la segmentation sémantique d'images aériennes. Nous évaluons notre approche sur le jeu de données exigeant FLAIR #2. Nous présentons une étude ablation exhaustive sur l'influence des différents composants de notre méthode sur la performance globale. Cette étude inclut une comparaison de différentes architectures de base (backbones), de techniques d'augmentation d'images, de stratégies de réglage du taux d'apprentissage, de fonctions de perte ainsi que de procédures d'entraînement. Nous proposons également une procédure d'entraînement en deux étapes et évaluons diverses options pour l'ensemblage de modèles. Sur la base de ces résultats, nous définissons la configuration finale du protocole d'entraînement. Cette configuration finale réduit l'erreur relative d'environ 18 % et atteint un mIoU de 0,641, constituant ainsi un nouveau record état de l'art. Notre code est disponible à l'adresse suivante : https://github.com/strakaj/U-Net-for-remote-sensing

Benchmarks

BenchmarkMéthodologieMétriques
semantic-segmentation-on-flair-french-landEnsemble-04 MiT-0 MiT-1 RNX-1 RNX-2
mIoU: 64.1

Construire l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec du co-codage IA gratuit, un environnement prêt à l'emploi et les meilleurs prix GPU.

Co-codage IA
GPU prêts à utiliser
Meilleurs prix
Commencer

Hyper Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp