🐦 multi_class_pigeon - Quickly annotate data on Jupyter

This repo is a simple multiclass image extenstion of the pigeon repo by @agermanidis located here. Credit goes to them for building the annotate function for images, text, and regression for multi-class type problems. All I am doing is extending it to multi-label and multi-task type problems.

Notes

As of now I just have this as a repo... so to use it you can clone it and either run notebooks from inside it and use the below format or modify the imports to point at the directory containing the annotation script.

Examples

from pigeon import multi_label_annotate
from IPython.display import display, Image

annotations = multi_label_annotate(
    ['assets/altera.jpg', 'assets/chibi_gil.jpg','assets/chibi_saber.jpg'],
    options={'cute':['yes','no'], 'saber':['yes','no'],'colors':['blue','gold','white','red']},
    display_fn=lambda filename: display(Image(filename))
    )

Preview:

Additional Notes

I added some additional buttons as well to the multi-label script. Since we want to be able to select more than one category I added a few extra buttons.

• done button to press after all relevant fields have been clicked.

• back button in case you want to go back to a previous image. WARNING this will erase that item from the dictionary so you will have to click through all the classes you want to mark for that previous image again.

• clear current will delete the current image from the dictionary on the back end so you can re enter the classes in case you messed one up.