heatMapSaver.hpp

Go to the documentation of this file.

#ifndef OPENPOSE_FILESTREAM_HEAT_MAP_SAVER_HPP
#define OPENPOSE_FILESTREAM_HEAT_MAP_SAVER_HPP
 
#include <openpose/core/common.hpp>
#include <openpose/filestream/fileSaver.hpp>
 
namespace op
{
    class OP_API HeatMapSaver : public FileSaver
    {
    public:
        HeatMapSaver(const std::string& directoryPath, const std::string& imageFormat);
 
        virtual ~HeatMapSaver();
 
        void saveHeatMaps(const std::vector<Array<float>>& heatMaps, const std::string& fileName) const;
 
    private:
        const std::string mImageFormat;
    };
}
 
#endif // OPENPOSE_FILESTREAM_HEAT_MAP_SAVER_HPP