detection_head.h File Reference

#include <array>
#include "opennn/core/opennn_types.h"

Go to the source code of this file.

Classes
struct	opennn::DetectionHeadMetadata
class	opennn::DetectionHeadEndpoint

Namespaces
namespace	opennn

Enumerations
enum class	opennn::DetectionHeadKind { opennn::AnchorBased , opennn::AnchorFree }
enum class	opennn::DetectionClassActivation { opennn::Softmax , opennn::Sigmoid }

Functions
float	opennn::yolo_box_iou (const std::array< float, 6 > &a, const std::array< float, 6 > &b)
float	opennn::dfl_decode (const float *logits, Index reg_max)