od_darknet_yolo_model_trainer.hpp File Reference

#include <ml/neural_net/compute_context.hpp>
#include <ml/neural_net/model_backend.hpp>
#include <ml/neural_net/model_spec.hpp>
#include <toolkits/object_detection/od_model_trainer.hpp>

Go to the source code of this file.

Classes
class	turi::object_detection::DarknetYOLOBackendTrainingWrapper
class	turi::object_detection::DarknetYOLOBackendInferenceWrapper
class	turi::object_detection::DarknetYOLOCheckpointer
class	turi::object_detection::DarknetYOLOCheckpoint
class	turi::object_detection::DarknetYOLOModelTrainer

Namespaces
	turi
	SKD.

Functions
neural_net::image_augmenter::options	turi::object_detection::DarknetYOLOInferenceAugmentationOptions (int batch_size, int output_height, int output_width)
neural_net::image_augmenter::options	turi::object_detection::DarknetYOLOTrainingAugmentationOptions (int batch_size, int output_height, int output_width)
EncodedInputBatch	turi::object_detection::EncodeDarknetYOLO (InputBatch input_batch, size_t output_height, size_t output_width, size_t num_anchors, size_t num_classes)
InferenceOutputBatch	turi::object_detection::DecodeDarknetYOLOInference (EncodedBatch batch, float confidence_threshold, float iou_threshold)

Detailed Description

Defines helper functions and the Model subclass for the darknet-yolo architecture.

Definition in file od_darknet_yolo_model_trainer.hpp.

Function Documentation

DarknetYOLOInferenceAugmentationOptions()

neural_net::image_augmenter::options turi::object_detection::DarknetYOLOInferenceAugmentationOptions	(	int	batch_size,
		int	output_height,
		int	output_width
	)

Configures an image_augmenter for inference given darknet-yolo network parameters.

DarknetYOLOTrainingAugmentationOptions()

neural_net::image_augmenter::options turi::object_detection::DarknetYOLOTrainingAugmentationOptions	(	int	batch_size,
		int	output_height,
		int	output_width
	)

Configures an image_augmenter for training given darknet-yolo network parameters.

DecodeDarknetYOLOInference()

InferenceOutputBatch turi::object_detection::DecodeDarknetYOLOInference	(	EncodedBatch	batch,
		float	confidence_threshold,
		float	iou_threshold
	)

Decodes the raw inference output into structured predictions.

EncodeDarknetYOLO()

EncodedInputBatch turi::object_detection::EncodeDarknetYOLO	(	InputBatch	input_batch,
		size_t	output_height,
		size_t	output_width,
		size_t	num_anchors,
		size_t	num_classes
	)

Encodes the annotations of an input batch into the format expected by the darknet-yolo network.