[PYTHON] Comment créer une étiquette (masque) pour la segmentation avec labelme (masque de segmentation sémantique)

↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓

Connaissances préalables

J'écris un article en supposant que j'ai les connaissances suivantes

• segmentation sémantique (# segmentation sémantique)
• mask（#label）

labelme est une application pour créer des étiquettes

C'est une application comme celle-ci

• labelme est une application de création d'étiquettes faite par python. Vous pouvez ajouter plusieurs étiquettes à votre image.
• Il n'y a aucun problème avec les autres classes. Donc, bien sûr, vous pouvez le faire avec 2 classes de segmentation sémantique.
• Tracez une ligne sur l'image avec des points et le résultat sera affiché sous forme de fichier JSON
• Si le fichier JSON est enregistré dans le même dossier que l'image avec le même nom (il est enregistré avec le même nom par défaut), le JSON sera chargé sans autorisation lorsque l'image est chargée, et il s'ouvrira dans un état annoté.
• Les points créés peuvent être à nouveau modifiés.

Détails sur l'utilisation de labelme

Télécharger label me

Pour les fenêtres

Pour Mac

Étiquette avec étiquette moi

Lorsque vous ouvrez labelme, la fenêtre suivante s'ouvre. (L'image est une capture d'écran de l'écran de recherche Google. C'est difficile à comprendre ...)

Tout en développant, tracez une ligne au point. Il est très facile de zoomer et dézoomer.

Je pense que c'est environ 1min avec ce sentiment approprié. Assez facile à utiliser. ..

Enregistrez le JSON de SAVE sur la gauche. Vous pouvez créer des données comme ça!

{
  "version": "4.5.6",
  "flags": {},
  "shapes": [
    {
      "label": "dog",
      "points": [
        [
          104.36893203883496,
          66.99029126213593
        ],
        [
          93.44660194174757,
          71.35922330097087
        ],
Omis ici
        [
          112.13592233009709,
          74.27184466019418
        ],
        [
          107.28155339805825,
          73.30097087378641
        ]
      ],
      "group_id": null,
      "shape_type": "polygon",
      "flags": {}
    }
  ],
  "imagePath": "Capture d'écran 2020-09-14 141400.png ",
  "imageData": "réduction",
  "imageHeight": 405,
  "imageWidth": 535
}

Mise en garde! Étant donné que ce point JSON est un point de polygone, il ne peut pas être utilisé comme masque tel quel. .. Nous devons donc convertir ce point en masque.

Convertir le fichier labelme de JSON en MASK

Au début, je pensais l'implémenter moi-même (c'est un problème), mais il y avait une fonction pour cela (c'est vrai). labelme a un module appelé shape.py dans le dossier utils, et vous pouvez convertir de JSON en MASK en utilisant la fonction shape_to_mask.

Cependant, comme il doit être implémenté subtilement, je publierai l'exemple de code.

import json
with open(path, "r",encoding="utf-8") as f:
    dj = json.load(f)
# dj['shapes'][0]Est pour une étiquette cette fois.
mask = shape_to_mask((dj['imageHeight'],dj['imageWidth']), dj['shapes'][0]['points'], shape_type=None,line_width=1, point_size=1)

mask_img = mask.astype(np.int)#booléen à 0,Convertir en 1

#J'utilise anaconda
import matplotlib.pyplot as plt
%matplotlib inline

plt.imshow(mask_img)

J'ai pu bien l'étiqueter. À ce niveau, il faut environ 2 à 3 minutes entre l'ouverture de l'image et son enregistrement, n'est-ce pas?

finalement

Veuillez me faire savoir si vous avez des outils d'étiquetage recommandés autres que m'étiqueter. Je me demande si certaines personnes l'ont mis en œuvre de leur propre chef.