Skip to content

Commit ed5a6c8

Browse files
committed
Adding test model into test data folder
1 parent 87e0559 commit ed5a6c8

File tree

5 files changed

+2
-2
lines changed

5 files changed

+2
-2
lines changed

cpp/tests/cppflow_model.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ TEST_CASE("Cppflow Model"){
3232
cppflow::tensor input_tensor = sopt::cppflowutils::convert_image_to_tensor(image, image_rows, image_cols);
3333

3434
// Read in model
35-
cppflow::model model(std::string(sopt::notinstalled::models_directory() + "/DnCNN/snr_15_model.pb/"));
35+
cppflow::model model(std::string(sopt::notinstalled::models_directory() + "/snr_15_model.pb/"));
3636

3737
// Run model on image
3838
// TODO: Automatically detect the string parameters, see issue #320
Binary file not shown.
Binary file not shown.
Binary file not shown.

cpp/tests/tf_inpainting.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ typedef sopt::Image<Scalar> Image;
3131
TEST_CASE("Inpainting"){
3232
extern std::unique_ptr<std::mt19937_64> mersenne;
3333
std::string const input = "cameraman256";
34-
std::string const model_path = std::string(sopt::notinstalled::models_directory() + "/DnCNN/snr_15_model.pb/");
34+
std::string const model_path = std::string(sopt::notinstalled::models_directory() + "/snr_15_model.pb/");
3535

3636
Image const image = sopt::notinstalled::read_standard_tiff(input);
3737

0 commit comments

Comments
 (0)