cocoJsonSaver.hpp

Go to the documentation of this file.

#ifndef OPENPOSE_FILESTREAM_POSE_JSON_COCO_SAVER_HPP
#define OPENPOSE_FILESTREAM_POSE_JSON_COCO_SAVER_HPP

#include <openpose/core/common.hpp>
#include <openpose/filestream/jsonOfstream.hpp>

namespace op
{
    class OP_API CocoJsonSaver
    {
    public:
        explicit CocoJsonSaver(const std::string& filePathToSave, const bool humanReadable = true);

        ~CocoJsonSaver();

        void record(const Array<float>& poseKeypoints, const Array<float>& poseScores, const std::string& imageName);

    private:
        JsonOfstream mJsonOfstream;
        bool mFirstElementAdded;

        DELETE_COPY(CocoJsonSaver);
    };
}

#endif // OPENPOSE_FILESTREAM_POSE_JSON_COCO_SAVER_HPP