#ifndef YOLO_H
#define YOLO_H

// Create the YOLOv8 image labels.

#include <iostream>
#include <fstream>

#include <opencv2/opencv.hpp>

// Name to assign to all labels.
#define LABEL_NAME "tooth"

typedef struct Label {
    int confidence;
    int xmin;
    int ymin;
    int xmax;
    int ymax;
} label_t;

label_t* yolo_mklabel(cv::Rect box);

// Write labels to file from cv::Rect.
void yolo_write_labels(std::string fname, std::vector<cv::Rect> boxes);


#endif
