4 releases

new 0.1.102 Nov 3, 2024
0.1.101 Nov 1, 2024
0.1.1 Oct 29, 2024
0.1.0 Oct 24, 2024

#150 in Machine learning

Download history 112/week @ 2024-10-22 274/week @ 2024-10-29

386 downloads per month

Apache-2.0

41KB
923 lines

yolo_io

A Rust library for loading, validating, and exporting YOLO project files.

Work in Progress

This crate is very much a work-in-progress. Features outlined may not be completed, working properly, or even listed.

Features

  1. Automatic pairing based on filename
    1. Pairing is simple; if three files exist, one image, three labels, it will list one valid pair and two errors. The valid pair will not be flagged as having possible other pairings.
  2. Flags incomplete pairs
  3. Checks for corrupt label files:
    1. Check for empty file
    2. Check for corrupted format
    3. Check if duplicates exist in the same file.
    4. Check if invalid class ids exist
    5. TODO: Compare labels in duplicate label files. What if they are different?
  4. Export YOLO project
    1. Unlike RectLabel, let's make it so there isn't other types of imports. We import a YOLO project, we export a YOLO project. Don't create a separate folder for annotations.
  5. Data quality validation report
    1. Exports a list of all labels with issues.

Reads

Dependencies

~3–4.5MB
~88K SLoC