Turi Create
4.0
|
#include <ml/neural_net/combine.hpp>
#include <ml/neural_net/compute_context.hpp>
#include <ml/neural_net/model_spec.hpp>
#include <toolkits/style_transfer/style_transfer_data_iterator.hpp>
Go to the source code of this file.
Namespaces | |
turi | |
SKD. | |
Functions | |
EncodedBatch | turi::style_transfer::EncodeTrainingBatch (DataBatch batch, int width, int height) |
EncodedInferenceBatch | turi::style_transfer::EncodeInferenceBatch (DataBatch batch) |
DataBatch | turi::style_transfer::DecodeInferenceBatch (EncodedInferenceBatch batch) |
Defines the value types representing each stage of a style-transfer training pipeline, and the virtual interface for arbitrary style-transfer models.
Definition in file st_model_trainer.hpp.
DataBatch turi::style_transfer::DecodeInferenceBatch | ( | EncodedInferenceBatch | batch | ) |
Converts the raw output from an inference backend into images.
EncodedInferenceBatch turi::style_transfer::EncodeInferenceBatch | ( | DataBatch | batch | ) |
Converts native images into tensors that can be fed into the model backend.
EncodedBatch turi::style_transfer::EncodeTrainingBatch | ( | DataBatch | batch, |
int | width, | ||
int | height | ||
) |
Converts native images into tensors that can be fed into the model backend.