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.

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:

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


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