Skip to main content

A way to label multi label image datasets in jupyter

Project description

🐦 multi_label_pigeon

This repo is a simple multiclass image extenstion of the pigeon library 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.


Can install with standard pip.

pip install multi-label-pigeon


from multi_label_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. back_example

  • clear current WARNING 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. clear_current

Project details

Download files

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

Source Distributions

No source distribution files available for this release. See tutorial on generating distribution archives.

Built Distribution

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page