Mirror/waifu2x-caffe
1
0
Fork 0
mirror of https://github.com/lltcggie/waifu2x-caffe.git synced 2025-06-26 05:32:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

新しいstbに対応

Browse Source
This commit is contained in:
lltcggie 2015-12-06 21:15:45 +09:00
parent b89194f1b8
commit f03f93e471
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
common/waifu2x.cpp
View File

@ -1184,6 +1184,12 @@ void Waifu2x::destroy()
is_inited = false; is_inited = false;
} }
static void Waifu2x_stbi_write_func(void *context, void *data, int size)
{
boost::iostreams::stream<boost::iostreams::file_descriptor> *osp = (boost::iostreams::stream<boost::iostreams::file_descriptor> *)context;
osp->write((const char *)data, size);
}
Waifu2x::eWaifu2xError Waifu2x::WriteMat(const cv::Mat &im, const boost::filesystem::path &output_file) Waifu2x::eWaifu2xError Waifu2x::WriteMat(const cv::Mat &im, const boost::filesystem::path &output_file)
{ {
const boost::filesystem::path ip(output_file); const boost::filesystem::path ip(output_file);
@ -1231,7 +1237,7 @@ Waifu2x::eWaifu2xError Waifu2x::WriteMat(const cv::Mat &im, const boost::filesys
if(!os) if(!os)
return eWaifu2xError_FailedOpenOutputFile; return eWaifu2xError_FailedOpenOutputFile;
if (!stbi_write_tga(os, im.size().width, im.size().height, im.channels(), data)) if (!stbi_write_tga_to_func(Waifu2x_stbi_write_func, &os, im.size().width, im.size().height, im.channels(), data))
return eWaifu2xError_FailedOpenOutputFile; return eWaifu2xError_FailedOpenOutputFile;
return eWaifu2xError_OK; return eWaifu2xError_OK;