labelr 0.1.0__py3-none-any.whl

labelr-0.1.0.dist-info/METADATA

@@ -0,0 +1,160 @@
+Metadata-Version: 2.1
+Name: labelr
+Version: 0.1.0
+Summary: Add your description here
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: datasets>=3.2.0
+Requires-Dist: imagehash>=4.3.1
+Requires-Dist: label-studio-sdk>=1.0.8
+Requires-Dist: more-itertools>=10.5.0
+Requires-Dist: openfoodfacts>=2.3.4
+Requires-Dist: protobuf>=5.29.1
+Requires-Dist: typer>=0.15.1
+Provides-Extra: ultralytics
+Requires-Dist: ultralytics>=8.3.49; extra == "ultralytics"
+Provides-Extra: triton
+Requires-Dist: tritonclient>=2.52.0; extra == "triton"
+
+# Labelr
+
+Labelr a command line interface that aims to provide a set of tools to help data scientists and machine learning engineers to deal with ML data annotation, data preprocessing and format conversion.
+
+This project started as a way to automate some of the tasks we do at Open Food Facts to manage data at different stages of the machine learning pipeline.
+
+The CLI currently is integrated with Label Studio (for data annotation), Ultralytics (for object detection) and Hugging Face (for model and dataset storage). It only works with some specific tasks (object detection only currently), but it's meant to be extended to other tasks in the future.
+
+It currently allows to:
+
+- create Label Studio projects
+- upload images to Label Studio
+- pre-annotate the tasks either with an existing object detection model run by Triton, or with Yolo-World (through Ultralytics)
+- perform data quality checks on Label Studio
+- export the data to Hugging Face Dataset or to local disk
+
+## Installation
+
+Python 3.10 or higher is required to run this CLI.
+You need to install the CLI manually for now, there is no project published on Pypi.
+To do so:
+
+We recommend to install the CLI in a virtual environment. You can either use pip or conda for that.
+
+### Pip
+
+Create the virtualenv:
+
+```bash
+python3 -m venv labelr
+source labelr/bin/activate
+```
+### Conda
+
+With conda:
+```bash
+conda create -n labelr python=3.12
+conda activate labelr
+```
+
+Then, clone the repository and install the requirements:
+
+```bash
+git clone git@github.com:openfoodfacts/openfoodfacts-ai.git
+```
+
+```bash
+pip install -r requirements.txt
+```
+
+We assume in the following that you have installed the CLI in a virtual environment, and defined the following alias in your shell configuration file (e.g. `.bashrc` or `.zshrc`):
+
+```bash
+alias labelr='${VIRTUALENV_DIR}/bin/python3 ${PROJECT_PATH}/main.py'
+```
+or if you are using conda:
+```bash
+alias labelr='${CONDA_PREFIX}/bin/python3 ${PROJECT_PATH}/main.py'
+``` 
+
+with `${VIRTUALENV_DIR}` the path to the virtual environment where you installed the CLI and `${PROJECT_PATH}` the path to the root of the project, for example:
+```bash
+${PROJECT_PATH} = /home/user/openfoodfacts-ai/ml_utils/ml_utils_cli
+```
+
+## Usage
+
+### Label Studio integration
+
+To create a Label Studio project, you need to have a Label Studio instance running. Launching a Label Studio instance is out of the scope of this project, but you can follow the instructions on the [Label Studio documentation](https://labelstud.io/guide/install.html).
+
+By default, the CLI will use Open Food Facts Label Studio instance, but you can change the URL by setting the `--label-studio-url` CLI option.
+
+For all the commands that interact with Label Studio, you need to provide an API key using the `--api-key` CLI option. You can get an API key by logging in to the Label Studio instance and going to the Account & Settings page.
+
+#### Create a project
+
+Once you have a Label Studio instance running, you can create a project with the following command:
+
+```bash
+labelr projects create --title my_project --api-key API_KEY --config-file label_config.xml
+```
+
+where `API_KEY` is the API key of the Label Studio instance (API key is available at Account page), and `label_config.xml` is the configuration file of the project.
+
+#### Create a dataset file
+
+If you have a list of images, for an object detection task, you can quickly create a dataset file with the following command:
+
+```bash
+labelr projects create-dataset-file --input-file image_urls.txt --output-file dataset.json
+```
+
+where `image_urls.txt` is a file containing the URLs of the images, one per line, and `dataset.json` is the output file.
+
+#### Import data
+
+Next, import the generated data to a project with the following command:
+
+```bash
+labelr projects import-data --project-id PROJECT_ID --dataset-path dataset.json
+```
+
+where `PROJECT_ID` is the ID of the project you created.
+
+#### Pre-annotate the data
+
+To accelerate annotation, you can pre-annotate the images with an object detection model. We support two pre-annotation backends:
+
+- Triton: you need to have a Triton server running with a model that supports object detection. The object detection model is expected to be a yolo-v8 model. You can set the URL of the Triton server with the `--triton-url` CLI option.
+
+- Ultralytics: you can use the [Yolo-World model from Ultralytics](https://github.com/ultralytics/ultralytics), Ultralytics should be installed in the same virtualenv.
+
+To pre-annotate the data with Triton, use the following command:
+
+```bash
+labelr projects add-prediction --project-id PROJECT_ID --backend ultralytics --labels 'product' --labels 'price tag' --label-mapping '{"price tag": "price-tag"}'
+```
+
+where `labels` is the list of labels to use for the object detection task (you can add as many labels as you want).
+For Ultralytics, you can also provide a `--label-mapping` option to map the labels from the model to the labels of the project.
+
+By default, for Ultralytics, the `yolov8x-worldv2.pt` model is used. You can change the model by setting the `--model-name` CLI option.
+
+#### Export the data
+
+Once the data is annotated, you can export it to a Hugging Face dataset or to local disk (Ultralytics format). To export it to disk, use the following command:
+
+```bash
+labelr datasets export --project-id PROJECT_ID --from ls --to ultralytics --output-dir output --label-names 'product,price-tag'
+```
+
+where `output` is the directory where the data will be exported. Currently, label names must be provided, as the CLI does not support exporting label names from Label Studio yet.
+
+To export the data to a Hugging Face dataset, use the following command:
+
+```bash
+labelr datasets export --project-id PROJECT_ID --from ls --to huggingface --repo-id REPO_ID --label-names 'product,price-tag'
+```
+
+where `REPO_ID` is the ID of the Hugging Face repository where the dataset will be uploaded (ex: `openfoodfacts/food-detection`).

labelr-0.1.0.dist-info/RECORD

@@ -0,0 +1,20 @@
+labelr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+labelr/__main__.py,sha256=G4e95-IfhI-lOmkOBP6kQ8wl1x_Fl7dZlLOYr90K83c,66
+labelr/annotate.py,sha256=8O9SO2thevo_Aa6etIUxCz2xJVXB4MwuSHj4jxz8sqQ,3441
+labelr/check.py,sha256=3wK6mE0UsKvoBNm0_lyWhCMq7gxkv5r50pvO70damXY,2476
+labelr/config.py,sha256=3RXF_NdkSuHvfVMGMlYmjlw45fU77zQkLX7gmZq7NxM,64
+labelr/export.py,sha256=tcOmVnOdJidWfNouNWoQ4OJgHMbbG-bLFHkId9huiS0,10170
+labelr/main.py,sha256=1_cZoJLBMpUV-lnaKb1XaVff4XxWjpIUZbSNQh44tPE,8715
+labelr/sample.py,sha256=cpzvgZWVU6GzwD35tqGKEFVKAgqQbSHlWW6IL9FG15Q,5918
+labelr/types.py,sha256=CahqnkLnGj23Jg0X9nftK7Jiorq50WYQqR8u9Ln4E-k,281
+labelr/apps/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+labelr/apps/datasets.py,sha256=DXU8XZx0iEHDI5SvUeI8atCKSUmj9YJwO6xTgMZDgEI,7936
+labelr/apps/projects.py,sha256=HpulSciBVTk1sSR1uXjtHytny9t-rN8wiaQ5llNBX6Y,12420
+labelr/apps/users.py,sha256=twQSlpHxE0hrYkgrJpEFbK8lYfWnpJr8vyfLHLtdAUU,909
+labelr/triton/object_detection.py,sha256=QKUOWiYFH72omyZH4SdbA56JDiVA_e_N8YCSQarkzWQ,7409
+labelr-0.1.0.dist-info/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+labelr-0.1.0.dist-info/METADATA,sha256=tBsu8c-LehNqjPNiCG3XjRLboQNeq2RSy9JZiv4v9Dc,6528
+labelr-0.1.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+labelr-0.1.0.dist-info/entry_points.txt,sha256=OACukVeR_2z54i8yQuWqqk_jdEHlyTwmTFOFBmxPp1k,43
+labelr-0.1.0.dist-info/top_level.txt,sha256=bjZo50aGZhXIcZYpYOX4sdAQcamxh8nwfEh7A9RD_Ag,7
+labelr-0.1.0.dist-info/RECORD,,

labelr-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (75.6.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

labelr-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+labelr = labelr.main:app

labelr-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+labelr