Loading...
Searching...
No Matches
yolo_dataset_internal.h File Reference

Go to the source code of this file.

Classes

struct  opennn::yolo_detail::YoloImageCacheHeader
 
struct  opennn::yolo_detail::YoloTargetCacheHeader
 
struct  opennn::yolo_detail::YoloBoxesCacheHeader
 
struct  opennn::yolo_detail::YoloBoxRecord
 
struct  opennn::yolo_detail::AnchorMatch
 

Namespaces

namespace  opennn
 
namespace  opennn::yolo_detail
 

Functions

float opennn::yolo_detail::clamp_unit (const float value)
 
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)
 
float opennn::yolo_detail::yolo_iou_wh (const array< float, 2 > &box, const array< float, 2 > &anchor)
 
Index opennn::yolo_detail::best_anchor_for_box (const YoloDataset::Box &box, const vector< array< float, 2 > > &anchors)
 
Index opennn::yolo_detail::grid_cell (float coordinate, Index grid)
 
void opennn::yolo_detail::write_box_target (float *cell, const YoloDataset::Box &box, Index grid, Index col, Index row, float objectness)
 
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)
 
AnchorMatch opennn::yolo_detail::best_anchor_across_heads (const YoloDataset::Box &box, const vector< vector< array< float, 2 > > > &head_anchors)
 
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)
 
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)
 
void opennn::yolo_detail::make_target_v8_gtlist (const vector< YoloDataset::Box > &boxes, Index classes_number, float *target)
 

Variables

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