TensorInfo Struct Reference

Detailed Description

Holds information about an output tensor of the yolo network.

Definition at line 51 of file yolo.h.

Data Fields
std::string	blobName
uint	stride {0}
uint	gridSize {0}
uint	numClasses {0}
uint	numBBoxes {0}
uint64_t	volume {0}
std::vector< uint >	masks
std::vector< float >	anchors
int	bindingIndex {-1}
float *	hostBuffer {nullptr}

Field Documentation

anchors

std::vector<float> TensorInfo::anchors

Definition at line 60 of file yolo.h.

bindingIndex

int TensorInfo::bindingIndex {-1}

Definition at line 61 of file yolo.h.

blobName

std::string TensorInfo::blobName

Definition at line 53 of file yolo.h.

gridSize

uint TensorInfo::gridSize {0}

Definition at line 55 of file yolo.h.

hostBuffer

float* TensorInfo::hostBuffer {nullptr}

Definition at line 62 of file yolo.h.

masks

std::vector<uint> TensorInfo::masks

Definition at line 59 of file yolo.h.

numBBoxes

uint TensorInfo::numBBoxes {0}

Definition at line 57 of file yolo.h.

numClasses

uint TensorInfo::numClasses {0}

Definition at line 56 of file yolo.h.

stride

uint TensorInfo::stride {0}

Definition at line 54 of file yolo.h.

volume

uint64_t TensorInfo::volume {0}

Definition at line 58 of file yolo.h.

---

The documentation for this struct was generated from the following file:

• yolo.h