Skip to main content

A way to label multi label image datasets in jupyter

Project description

🐦 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.


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.


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))


alt text

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.

Project details

Release history Release notifications

This version


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for multi-label-pigeon-jupyter, version 0.1.0
Filename, size File type Python version Upload date Hashes
Filename, size multi_label_pigeon_jupyter-0.1.0-py3-none-any.whl (4.6 kB) File type Wheel Python version py3 Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page