Skip to main content

You Actually Look Twice At it, YOLOv5-Kraken adapter for region detection

Project description

YALTAi

You Actually Look Twice At it

This provides an adapter for Kraken to use YOLOv8 (1.0.0 update; use previous version to reuse YOLOv5 models) Object Detection routine.

This tool can be used for both segmenting and conversion of models.

Routine

Instal

pip install YALTAi

Training

Convert (and split optionally) your data

# Keeps .1 data in the validation set and convert all alto into YOLOv5 format
#  Keeps the segmonto information up to the regions
yaltai convert alto-to-yolo PATH/TO/ALTOorPAGE/*.xml my-dataset --shuffle .1 --segmonto region

And then train YOLO

yolo task=detect mode=train model=yolov8n.pt data=my-dataset/config.yml epochs=100 plots=True device=0 batch=8 imgsz=960

Predicting

YALTAi has the same CLI interface as Kraken, so:

  • You can use base BLLA model for line or provide yours with -i model.mlmodel
  • Use a GPU (--device cuda:0) or a CPU (--device cpu)
  • Apply on batch (*.jpg)
# Retrieve the best.pt after the training
# It should be in runs/train/exp[NUMBER]/weights/best.pt
# And then annotate your new data with the same CLI API as Kraken !
yaltai kraken --device cuda:0 -I "*.jpg" --suffix ".xml" segment --yolo runs/train/exp5/weights/best.pt

Metrics

The metrics produced from various libraries never gives the same mAP or Precision. I tried

  • object-detection-metrics==0.4
  • mapCalc
  • mean-average-precision which ended up being the chosen one (cleanest in terms of how I can access info)

and of course I compared with YOLOv5 raw results. Nothing worked the same. And the library YOLOv5 derives its metrics from is uninstallable through pip.

Project details


Download files

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

Source Distribution

YALTAi-2.0.2.tar.gz (26.9 kB view details)

Uploaded Source

Built Distribution

YALTAi-2.0.2-py2.py3-none-any.whl (40.0 kB view details)

Uploaded Python 2 Python 3

File details

Details for the file YALTAi-2.0.2.tar.gz.

File metadata

  • Download URL: YALTAi-2.0.2.tar.gz
  • Upload date:
  • Size: 26.9 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.10.12

File hashes

Hashes for YALTAi-2.0.2.tar.gz
Algorithm Hash digest
SHA256 7af2a9420b32d4131313c3fd046ff6688861fea58baa1c0acd00fe039e522e2d
MD5 d2f808af199c8aebda82d37f1bfcbcd4
BLAKE2b-256 90947583dbdbdc97af5b7d8ae2999b4288deaea54b443e6275a5048b88c689c7

See more details on using hashes here.

File details

Details for the file YALTAi-2.0.2-py2.py3-none-any.whl.

File metadata

  • Download URL: YALTAi-2.0.2-py2.py3-none-any.whl
  • Upload date:
  • Size: 40.0 kB
  • Tags: Python 2, Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.10.12

File hashes

Hashes for YALTAi-2.0.2-py2.py3-none-any.whl
Algorithm Hash digest
SHA256 9ec7d2d0000351563e757f92b6493f6cb9d0726b20e749937be4bd6ea8eba7bb
MD5 4f6c9d6e374a29c4ffd17399d83c0db9
BLAKE2b-256 a97e9aca54252aa6c500595f776742a6d73c3ce4bcb4d2659256e5cf3b100cc6

See more details on using hashes here.

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