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.

Built Distribution

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page