turi::object_detection::DarknetYOLOModelTrainer Class Reference

#include <toolkits/object_detection/od_darknet_yolo_model_trainer.hpp>

Public Member Functions
	DarknetYOLOModelTrainer (const DarknetYOLOCheckpoint &checkpoint, neural_net::compute_context *context)
std::shared_ptr< neural_net::Publisher< TrainingOutputBatch > >	AsTrainingBatchPublisher (std::unique_ptr< data_iterator > training_data, size_t batch_size, int offset) override
std::shared_ptr< neural_net::Publisher< EncodedBatch > >	AsInferenceBatchPublisher (std::unique_ptr< data_iterator > test_data, size_t batch_size, float confidence_threshold, float iou_threshold) override
InferenceOutputBatch	DecodeOutputBatch (EncodedBatch batch, float confidence_threshold, float iou_threshold) override
std::shared_ptr< neural_net::Publisher< std::unique_ptr< Checkpoint > > >	AsCheckpointPublisher () override

Detailed Description

Subclass of ModelTrainer encapsulating the darknet-yolo architecture.

Definition at line 166 of file od_darknet_yolo_model_trainer.hpp.

Constructor & Destructor Documentation

DarknetYOLOModelTrainer()

turi::object_detection::DarknetYOLOModelTrainer::DarknetYOLOModelTrainer	(	const DarknetYOLOCheckpoint &	checkpoint,
		neural_net::compute_context *	context
	)

Initializes a model from a checkpoint.

Member Function Documentation

AsCheckpointPublisher()

std::shared_ptr<neural_net::Publisher<std::unique_ptr<Checkpoint> > > turi::object_detection::DarknetYOLOModelTrainer::AsCheckpointPublisher ( )

overridevirtual

Returns a publisher that can be used to request checkpoints.

Implements turi::object_detection::ModelTrainer.

AsInferenceBatchPublisher()

std::shared_ptr<neural_net::Publisher<EncodedBatch> > turi::object_detection::DarknetYOLOModelTrainer::AsInferenceBatchPublisher ( std::unique_ptr< data_iterator >  test_data, size_t  batch_size, float  confidence_threshold, float  iou_threshold  )

overridevirtual

Given a data iterator, return a publisher of inference model outputs.

Implements turi::object_detection::ModelTrainer.

AsTrainingBatchPublisher()

std::shared_ptr<neural_net::Publisher<TrainingOutputBatch> > turi::object_detection::DarknetYOLOModelTrainer::AsTrainingBatchPublisher ( std::unique_ptr< data_iterator >  training_data, size_t  batch_size, int  offset  )

overridevirtual

Given a data iterator, return a publisher of model outputs.

Reimplemented from turi::object_detection::ModelTrainer.

DecodeOutputBatch()

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

overridevirtual

Convert the raw output of the inference batch publisher into structured predictions.

Implements turi::object_detection::ModelTrainer.

---

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

• toolkits/object_detection/od_darknet_yolo_model_trainer.hpp