StephanAkkerman/chart-info-detector

🤗 Hugging Face sourceobject-detectionmit35 MBother✓ 2 checksumsupdated today

Submit in one command

Run it next to your model folder. It makes the torrent, checks your files against Hugging Face, and submits it. You just start seeding and paste your key from your account. It only reads your files and never changes them. Read the script first if you like.

curl -fsSL https://pirateface.co/package.sh | bash -s -- --repo StephanAkkerman/chart-info-detector ./model-folder

Needs a seeder →

Chart Info Detector

This is a fintuned model for detecting objects in financial charts. It uses YOLO12n as its base model, making it a fast and small model. This model is trained on my own datasets of financial charts posted on Twitter, which I labeled myself.

Inteded uses

chart-info-detector is inteded for finding relevant information from financial chart images.

An example of a labelled financial chart:

Usage

To use this model you need to install the ultralytics library with Python. You can then download the weights of this repo and load them into the model. The image size during training was set to 1792, so be sure to use this too.

from ultralytics import YOLO
from huggingface_hub import hf_hub_download

# Load the pre-trained model
model = YOLO(hf_hub_download(
    repo_id="StephanAkkerman/chart-info-detector",
    filename="weights/best.pt",
    repo_type="model",
))

# Perform object detection on an image
results = model.predict(
    source=img_path,
    imgsz=1792,
    conf=0.25,
    iou=0.5,
    verbose=False,
)

r = results[0]

# Show the results 
r.show()