duan8/yolov10/include/types.h
mpj1234 17d61f4b6e
TensorRT8 support for YOLOv10 (#1561)
* TensorRT8 support for YOLOv10

* delelet images,  modify add test images in readme
2024-07-29 16:16:09 +08:00

17 lines
356 B
C

#pragma once
#include "config.h"
struct alignas(float) Detection {
//center_x center_y w h
float bbox[4];
float conf; // bbox_conf * cls_conf
float class_id;
};
struct AffineMatrix {
float value[6];
};
const int bbox_element =
sizeof(Detection) / sizeof(float) + 1; // left, top, right, bottom, confidence, class, keepflag