Turi Create  4.0
turi::object_detection::data_iterator::parameters Struct Reference

#include <toolkits/object_detection/od_data_iterator.hpp>

Public Attributes

gl_sframe data
 
std::string annotations_column_name
 
std::string predictions_column_name
 
std::string image_column_name
 
std::vector< std::string > class_labels
 
bool repeat = true
 
bool shuffle = true
 
bool is_training = false
 
int random_seed = 0
 

Detailed Description

Defines the inputs to a data_iterator factory function.

Definition at line 51 of file od_data_iterator.hpp.

Member Data Documentation

◆ annotations_column_name

std::string turi::object_detection::data_iterator::parameters::annotations_column_name

The name of the column containing the annotations.

The values must either be dictionaries containing an annotation, or a list of such dictionaries. An annotation dictionary has a "label" key whose value is a string, and a "coordinates" key whose value is another dictionary containing "x", "y", "width", and "height", describing the position of the center and the size of the bounding box (in the image's coordinates, with the origin at the top left).

Definition at line 66 of file od_data_iterator.hpp.

◆ class_labels

std::vector<std::string> turi::object_detection::data_iterator::parameters::class_labels

The expected class labels, indexed by identifier.

If empty, then the labels will be inferred from the data. If non-empty, an exception will be thrown upon encountering an unexpected label.

: This should be a flex_list to accomodate integer labels!

Definition at line 91 of file od_data_iterator.hpp.

◆ data

gl_sframe turi::object_detection::data_iterator::parameters::data

The SFrame to traverse

Definition at line 54 of file od_data_iterator.hpp.

◆ image_column_name

std::string turi::object_detection::data_iterator::parameters::image_column_name

The name of the column containing the images.

Each value is either an image or a path to an image file on disk.

Definition at line 81 of file od_data_iterator.hpp.

◆ is_training

bool turi::object_detection::data_iterator::parameters::is_training = false

Whether the process is in training or not.

Definition at line 106 of file od_data_iterator.hpp.

◆ predictions_column_name

std::string turi::object_detection::data_iterator::parameters::predictions_column_name

Optional name of a column containing predictions.

If not empty, then the iterator will parse and yield a secondary stream of bounding boxes, intended for use in evaluating existing predictions.

Definition at line 74 of file od_data_iterator.hpp.

◆ random_seed

int turi::object_detection::data_iterator::parameters::random_seed = 0

Determines results of shuffle operations if enabled.

Definition at line 109 of file od_data_iterator.hpp.

◆ repeat

bool turi::object_detection::data_iterator::parameters::repeat = true

Whether to traverse the data more than once.

Definition at line 100 of file od_data_iterator.hpp.

◆ shuffle

bool turi::object_detection::data_iterator::parameters::shuffle = true

Whether to shuffle the data on subsequent traversals.

Definition at line 103 of file od_data_iterator.hpp.


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