Loading...
Searching...
No Matches
opennn::yolo_detail Namespace Reference

Classes

struct  AnchorMatch
 
struct  YoloBoxesCacheHeader
 
struct  YoloBoxRecord
 
struct  YoloImageCacheHeader
 
struct  YoloTargetCacheHeader
 

Functions

float clamp_unit (const float value)
 
void append_mosaic_boxes (span< const YoloDataset::Box > source, const array< float, 2 > &tile_origin, const array< float, 2 > &tile_size, vector< YoloDataset::Box > &destination)
 
float yolo_iou_wh (const array< float, 2 > &box, const array< float, 2 > &anchor)
 
Index best_anchor_for_box (const YoloDataset::Box &box, const vector< array< float, 2 > > &anchors)
 
Index grid_cell (float coordinate, Index grid)
 
void write_box_target (float *cell, const YoloDataset::Box &box, Index grid, Index col, Index row, float objectness)
 
void mark_ignored_anchors (const YoloDataset::Box &box, const vector< array< float, 2 > > &anchors, Index grid, Index head_offset, Index values_per_box, Index head_channels, Index skip_anchor, float *target)
 
AnchorMatch best_anchor_across_heads (const YoloDataset::Box &box, const vector< vector< array< float, 2 > > > &head_anchors)
 
void make_target (const vector< YoloDataset::Box > &boxes, const vector< array< float, 2 > > &anchors, Index grid_size, Index boxes_per_cell, Index classes_number, float *target)
 
void make_target_multi_scale (const vector< YoloDataset::Box > &boxes, const vector< vector< array< float, 2 > > > &head_anchors, const vector< Index > &head_grid_sizes, Index boxes_per_head, Index classes_number, float *target)
 
void make_target_v8_gtlist (const vector< YoloDataset::Box > &boxes, Index classes_number, float *target)
 

Variables

constexpr uint32_t YOLO_CACHE_VERSION = 3
 
constexpr char YOLO_IMAGE_MAGIC [8] = {'O','P','E','N','N','Y','I','M'}
 
constexpr char YOLO_TARGET_MAGIC [8] = {'O','P','E','N','N','Y','T','G'}
 
constexpr char YOLO_BOXES_MAGIC [8] = {'O','P','E','N','N','Y','B','X'}
 

Function Documentation

◆ append_mosaic_boxes()

void opennn::yolo_detail::append_mosaic_boxes ( span< const YoloDataset::Box > source,
const array< float, 2 > & tile_origin,
const array< float, 2 > & tile_size,
vector< YoloDataset::Box > & destination )
inline

◆ best_anchor_across_heads()

AnchorMatch opennn::yolo_detail::best_anchor_across_heads ( const YoloDataset::Box & box,
const vector< vector< array< float, 2 > > > & head_anchors )
inline

◆ best_anchor_for_box()

Index opennn::yolo_detail::best_anchor_for_box ( const YoloDataset::Box & box,
const vector< array< float, 2 > > & anchors )
inline

◆ clamp_unit()

float opennn::yolo_detail::clamp_unit ( const float value)
inline

◆ grid_cell()

Index opennn::yolo_detail::grid_cell ( float coordinate,
Index grid )
inline

◆ make_target()

void opennn::yolo_detail::make_target ( const vector< YoloDataset::Box > & boxes,
const vector< array< float, 2 > > & anchors,
Index grid_size,
Index boxes_per_cell,
Index classes_number,
float * target )
inline

◆ make_target_multi_scale()

void opennn::yolo_detail::make_target_multi_scale ( const vector< YoloDataset::Box > & boxes,
const vector< vector< array< float, 2 > > > & head_anchors,
const vector< Index > & head_grid_sizes,
Index boxes_per_head,
Index classes_number,
float * target )
inline

◆ make_target_v8_gtlist()

void opennn::yolo_detail::make_target_v8_gtlist ( const vector< YoloDataset::Box > & boxes,
Index classes_number,
float * target )
inline

◆ mark_ignored_anchors()

void opennn::yolo_detail::mark_ignored_anchors ( const YoloDataset::Box & box,
const vector< array< float, 2 > > & anchors,
Index grid,
Index head_offset,
Index values_per_box,
Index head_channels,
Index skip_anchor,
float * target )
inline

◆ write_box_target()

void opennn::yolo_detail::write_box_target ( float * cell,
const YoloDataset::Box & box,
Index grid,
Index col,
Index row,
float objectness )
inline

◆ yolo_iou_wh()

float opennn::yolo_detail::yolo_iou_wh ( const array< float, 2 > & box,
const array< float, 2 > & anchor )
inline

Variable Documentation

◆ YOLO_BOXES_MAGIC

char opennn::yolo_detail::YOLO_BOXES_MAGIC[8] = {'O','P','E','N','N','Y','B','X'}
constexpr

◆ YOLO_CACHE_VERSION

uint32_t opennn::yolo_detail::YOLO_CACHE_VERSION = 3
constexpr

◆ YOLO_IMAGE_MAGIC

char opennn::yolo_detail::YOLO_IMAGE_MAGIC[8] = {'O','P','E','N','N','Y','I','M'}
constexpr

◆ YOLO_TARGET_MAGIC

char opennn::yolo_detail::YOLO_TARGET_MAGIC[8] = {'O','P','E','N','N','Y','T','G'}
constexpr