Command Palette
Search for a command to run...
Amorçage de la segmentation sémantique par contraste régional
Amorçage de la segmentation sémantique par contraste régional
Shikun Liu Shuaifeng Zhi Edward Johns Andrew J. Davison
Résumé
Nous présentons ReCo, un cadre d'apprentissage contrastif conçu à l'échelle régionale afin d'assister l'apprentissage en segmentation sémantique. ReCo effectue un apprentissage contrastif au niveau des pixels, supervisé ou semi-supervisé, sur un ensemble éparse de pixels négatifs difficiles, avec une surcharge mémoire négligeable. ReCo est facile à implémenter, puisqu'il repose sur des réseaux de segmentation standards disponibles commercialement, et améliore de manière cohérente les performances des méthodes de segmentation sémantique supervisées comme semi-supervisées, en produisant des frontières de segmentation plus lisses et une convergence plus rapide. Son effet le plus marqué se manifeste dans l'apprentissage semi-supervisé avec très peu d'étiquettes. Grâce à ReCo, nous parvenons à des modèles de segmentation sémantique de haute qualité, nécessitant uniquement 5 exemples par classe sémantique. Le code est disponible à l'adresse suivante : https://github.com/lorenmt/reco.