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 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.4mapCalcmean-average-precisionwhich 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.
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for YALTAi-1.0.2-py2.py3-none-any.whl
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 03a36fa720c9ddeaba04506235b60f67e49fa9a677923383b182bd84b6db1762 |
|
| MD5 | bb1259ec1d0c38a488adbcc27a7b06c8 |
|
| BLAKE2b-256 | dad675cb29e4066bd98142dbad97a9b41a6a13858f6352139994df438e3b013b |
