e73bffcd25
* tensorrt-yolov9 * format change * change format * Update block.cpp * format: add space * update block.cpp: add space * update config.h --------- Co-authored-by: Wang Xinyu <wangxinyu_es@163.com>
18 lines
360 B
C
18 lines
360 B
C
#pragma once
|
|
|
|
#include "config.h"
|
|
|
|
struct YoloKernel {
|
|
int width;
|
|
int height;
|
|
float anchors[kNumAnchor * 2];
|
|
};
|
|
|
|
struct alignas(float) Detection {
|
|
float bbox[4]; // center_x center_y w h
|
|
float conf; // bbox_conf * cls_conf
|
|
float class_id;
|
|
float mask[32];
|
|
};
|
|
const int bbox_element = 7; // center_x, center_y, w, h, conf, cls, obj
|