#ifndef _YOLO_CONFIGS_H_
#define _YOLO_CONFIGS_H_


namespace Yolo
{
    static constexpr int CHECK_COUNT = 3;
    static constexpr float IGNORE_THRESH = 0.5f;
    static constexpr int CLASS_NUM = 6;

    struct YoloKernel
    {
        int width;
        int height;
        float anchors[CHECK_COUNT*2];
    };

    // //YOLO 608
    // YoloKernel yolo1 = {
    //     19,
    //     19,
    //     {116,90,  156,198,  373,326}
    // };
    // YoloKernel yolo2 = {
    //     38,
    //     38,
    //     {30,61,  62,45,  59,119}
    // };
    // YoloKernel yolo3 = {
    //     76,
    //     76,
    //     {10,13,  16,30,  33,23}
    // };

	// 

    //  //YOLO 416
    YoloKernel yolo1 = {
        13,
        13,
        {81,82,  135,169,  344,319}
    };
    YoloKernel yolo2 = {
        26,
        26,
        {10,14,  23,27,  37,58}
    };

    //YOLO 416
    // YoloKernel yolo1 = {
    //     13,
    //     13,
    //     {116,90,  156,198,  373,326}
    // };
    // YoloKernel yolo2 = {
    //     26,
    //     26,
    //     {30,61,  62,45,  59,119}
    // };
    // YoloKernel yolo3 = {
    //     52,
    //     52,
    //     {10,13,  16,30,  33,23}
    // };
}

#endif