diff --git a/bin/models/anime_style_art/srcnn.prototxt b/bin/models/anime_style_art/scale2.0x_model.prototxt similarity index 100% rename from bin/models/anime_style_art/srcnn.prototxt rename to bin/models/anime_style_art/scale2.0x_model.prototxt diff --git a/bin/models/anime_style_art/srcnn.prototxt.protobin b/bin/models/anime_style_art/srcnn.prototxt.protobin deleted file mode 100644 index a8db633..0000000 Binary files a/bin/models/anime_style_art/srcnn.prototxt.protobin and /dev/null differ diff --git a/bin/models/anime_style_art_rgb/srcnn.prototxt b/bin/models/anime_style_art_rgb/scale2.0x_model.prototxt similarity index 100% rename from bin/models/anime_style_art_rgb/srcnn.prototxt rename to bin/models/anime_style_art_rgb/scale2.0x_model.prototxt diff --git a/bin/models/anime_style_art_rgb/srcnn.prototxt.protobin b/bin/models/anime_style_art_rgb/srcnn.prototxt.protobin deleted file mode 100644 index 1432d8d..0000000 Binary files a/bin/models/anime_style_art_rgb/srcnn.prototxt.protobin and /dev/null differ diff --git a/bin/models/photo/srcnn.prototxt.protobin b/bin/models/photo/srcnn.prototxt.protobin deleted file mode 100644 index 1432d8d..0000000 Binary files a/bin/models/photo/srcnn.prototxt.protobin and /dev/null differ diff --git a/bin/models/ukbench/srcnn.prototxt.protobin b/bin/models/ukbench/srcnn.prototxt.protobin deleted file mode 100644 index 1432d8d..0000000 Binary files a/bin/models/ukbench/srcnn.prototxt.protobin and /dev/null differ diff --git a/bin/models/upconv_7_anime_style_art_rgb/noise1_model.json.caffemodel b/bin/models/upconv_7_anime_style_art_rgb/noise1_model.json.caffemodel index 3585aba..8dfaf9e 100644 Binary files a/bin/models/upconv_7_anime_style_art_rgb/noise1_model.json.caffemodel and b/bin/models/upconv_7_anime_style_art_rgb/noise1_model.json.caffemodel differ diff --git a/bin/models/upconv_7_anime_style_art_rgb/noise2_model.json.caffemodel b/bin/models/upconv_7_anime_style_art_rgb/noise2_model.json.caffemodel deleted file mode 100644 index 2fc81c9..0000000 Binary files a/bin/models/upconv_7_anime_style_art_rgb/noise2_model.json.caffemodel and /dev/null differ diff --git a/bin/models/upconv_7_anime_style_art_rgb/noise3_model.json.caffemodel b/bin/models/upconv_7_anime_style_art_rgb/noise3_model.json.caffemodel deleted file mode 100644 index 49eb245..0000000 Binary files a/bin/models/upconv_7_anime_style_art_rgb/noise3_model.json.caffemodel and /dev/null differ diff --git a/bin/models/upconv_7_anime_style_art_rgb/scale2.0x_model.json.caffemodel b/bin/models/upconv_7_anime_style_art_rgb/scale2.0x_model.json.caffemodel index eb98838..1acb14b 100644 Binary files a/bin/models/upconv_7_anime_style_art_rgb/scale2.0x_model.json.caffemodel and b/bin/models/upconv_7_anime_style_art_rgb/scale2.0x_model.json.caffemodel differ diff --git a/bin/models/upconv_7_anime_style_art_rgb/srcnn.prototxt b/bin/models/upconv_7_anime_style_art_rgb/scale2.0x_model.prototxt similarity index 100% rename from bin/models/upconv_7_anime_style_art_rgb/srcnn.prototxt rename to bin/models/upconv_7_anime_style_art_rgb/scale2.0x_model.prototxt diff --git a/bin/models/upconv_7_anime_style_art_rgb/srcnn.prototxt.protobin b/bin/models/upconv_7_anime_style_art_rgb/srcnn.prototxt.protobin deleted file mode 100644 index a4165b1..0000000 Binary files a/bin/models/upconv_7_anime_style_art_rgb/srcnn.prototxt.protobin and /dev/null differ diff --git a/common/waifu2x.cpp b/common/waifu2x.cpp index 2a71fbc..b455114 100644 --- a/common/waifu2x.cpp +++ b/common/waifu2x.cpp @@ -978,7 +978,7 @@ Waifu2x::eWaifu2xError Waifu2x::LoadInfoFromJson(const boost::filesystem::path & } // ネットワークを使って画像を再構築する -Waifu2x::eWaifu2xError Waifu2x::ReconstructImage(boost::shared_ptr> net, cv::Mat &im) +Waifu2x::eWaifu2xError Waifu2x::ReconstructImage(boost::shared_ptr> net, const int reconstructed_scale, cv::Mat &im) { const auto Height = im.size().height; const auto Width = im.size().width; @@ -989,7 +989,7 @@ Waifu2x::eWaifu2xError Waifu2x::ReconstructImage(boost::shared_ptr image_size = im.size() * inner_scale; + cv::Size_ image_size = im.size(); if (isReconstructNoise) { PaddingImage(im, im); - ret = ReconstructImage(net_noise, im); + ret = ReconstructImage(net_noise, 1, im); if (ret != eWaifu2xError_OK) return ret; @@ -1565,6 +1569,7 @@ Waifu2x::eWaifu2xError Waifu2x::ReconstructFloatMat(const bool isReconstructNois if (cancel_func && cancel_func()) return eWaifu2xError_Cancel; + image_size = im.size() * inner_scale; const double ratio = CalcScaleRatio(image_size); const int scale2 = ceil(log(ratio) / log(model_scale)); @@ -1585,7 +1590,7 @@ Waifu2x::eWaifu2xError Waifu2x::ReconstructFloatMat(const bool isReconstructNois PaddingImage(im, im); } - ret = ReconstructImage(net_scale, im); + ret = ReconstructImage(net_scale, inner_scale, im); if (ret != eWaifu2xError_OK) return ret; diff --git a/common/waifu2x.h b/common/waifu2x.h index 6ab2ece..0f20855 100644 --- a/common/waifu2x.h +++ b/common/waifu2x.h @@ -135,7 +135,7 @@ private: , const boost::filesystem::path &modelbin_path, const boost::filesystem::path &caffemodel_path, const std::string &process); eWaifu2xError LoadInfoFromJson(const boost::filesystem::path &info_path); eWaifu2xError SetParameter(caffe::NetParameter ¶m, const std::string &process) const; - eWaifu2xError ReconstructImage(boost::shared_ptr> net, cv::Mat &im); + eWaifu2xError ReconstructImage(boost::shared_ptr> net, const int reconstructed_scale, cv::Mat &im); static eWaifu2xError WriteMat(const cv::Mat &im, const boost::filesystem::path &output_file, const boost::optional &output_quality); eWaifu2xError BeforeReconstructFloatMatProcess(const cv::Mat &in, cv::Mat &out, bool &convertBGRflag);