data_preparation.hpp

/* Copyright © 2019 Apple Inc. All rights reserved.
 *
 * Use of this source code is governed by a BSD-3-clause license that can
 * be found in the LICENSE.txt file or at https://opensource.org/licenses/BSD-3-Clause
 */
#ifndef TURI_DRAWING_DATA_PREPARATION_H_
#define TURI_DRAWING_DATA_PREPARATION_H_

#include <core/export.hpp>
#include <core/data/sframe/gl_sframe.hpp>
#include <model_server/lib/image_util.hpp>

namespace turi {
namespace drawing_classifier {

/**
 *  This function is responsible for converting stroke-based drawing data
 *  to 28x28 bitmaps to make it ready for training with the Neural Network.
 *
 *
 * \param[in] data: SFrame from the user, which contains stroke-based drawings.
 *          Every stroke-based drawing must be represented as
 *          a list of strokes and each stroke must be represented as
 *          a list of points. Each point must be a dictionary with
 *          exactly two keys, "x" and "y".
 * \param[in] feature: Name of the feature column
 *
 *
 * \return: SFrame with the stroke-based drawings converted to bitmaps, which
 *      are represented as grayscale tc.Image of size 28x28.
 */
EXPORT gl_sframe _drawing_classifier_prepare_data(const gl_sframe &data,
                          const std::string &feature);


}
}

#endif //TURI_DRAWING_DATA_PREPARATION_H_