Detectron2

La bibliothèque PyTorch de Meta AI pour la détection et la segmentation d'instances.

Detectron2 est la bibliothèque de détection d'objets et de segmentation de Meta AI, écrite en PyTorch : son model zoo fournit Faster R-CNN, Mask R-CNN, RetinaNet, Cascade R-CNN, Panoptic FPN et Keypoint R-CNN pré-entraînés sur COCO. Un dataset au format COCO s'enregistre dans le DatasetCatalog, la configuration YAML ou LazyConfig pilote l'entraînement, et les modèles s'exportent en TorchScript. C'est notre choix quand il faut des masques d'instances ou une segmentation panoptique de précision, plutôt que la seule vitesse d'un détecteur temps réel.