Loading...
Searching...
No Matches
yolo_dataset.h File Reference
#include <cstdio>
#include "opennn/dataset/dataset.h"
#include "opennn/core/io_utilities.h"

Go to the source code of this file.

Classes

struct  opennn::YoloDetection
 
struct  opennn::YoloFpnHead
 
class  opennn::YoloDataset
 
struct  opennn::YoloDataset::Box
 
struct  opennn::YoloDataset::AugmentationPolicy
 

Namespaces

namespace  opennn
 

Functions

vector< YoloDetection > opennn::decode_yolo_detections (span< const float >, Index, Index, Index, Index)
 
vector< YoloDetection > opennn::decode_yolo_fpn_detections (const vector< YoloFpnHead > &, Index, Index, Index, Index, float confidence_threshold=0.25f, float iou_threshold=0.45f)
 
vector< YoloDetection > opennn::decode_yolo_v8_fpn_detections (const vector< YoloFpnHead > &, Index, Index, Index, Index, float confidence_threshold=0.25f, float iou_threshold=0.45f, Index reg_max=1)