Turi Create
4.0
|
#include <toolkits/object_detection/od_model_trainer.hpp>
Public Attributes | |
int | max_iterations = -1 |
int | batch_size = -1 |
int | output_height = 13 |
int | output_width = 13 |
int | num_classes = -1 |
Ostensibly model-agnostic parameters for object detection.
Definition at line 108 of file od_model_trainer.hpp.
int turi::object_detection::Config::batch_size = -1 |
The number of images to process per training batch.
If -1, then this size should be computed automatically.
Definition at line 121 of file od_model_trainer.hpp.
int turi::object_detection::Config::max_iterations = -1 |
The target number of training iterations to perform.
If -1, then this target should be computed heuristically.
Definition at line 114 of file od_model_trainer.hpp.
int turi::object_detection::Config::num_classes = -1 |
Determines the number of feature channels in the final feature map.
Definition at line 130 of file od_model_trainer.hpp.
int turi::object_detection::Config::output_height = 13 |
For darknet-yolo, the height of the final feature map.
Definition at line 124 of file od_model_trainer.hpp.
int turi::object_detection::Config::output_width = 13 |
For darknet-yolo, the width of the final feature map.
Definition at line 127 of file od_model_trainer.hpp.