#pragma once

#include "yolo_runtime.h"

#ifdef __cplusplus
extern "C" {
#endif

int write_result_json(
    const char* path,
    const char* model_config,
    const yolo_result_set_t* results,
    yolo_task_type_t task);

int write_result_json_masks(const char* path, const char* model_config,
    const yolo_result_set_t* results, yolo_task_type_t task,
    const yolo_mask_view_v1_t* prepared_masks, int dilation_radius);

#ifdef __cplusplus
}
#endif