Browse Source

!2982 supporting cpp unit tests for randomresizewithbbox and resizewithbbox

Merge pull request !2982 from ava/CPP_test3
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
2c3c204b14
5 changed files with 115 additions and 0 deletions
  1. +2
    -0
      tests/ut/cpp/dataset/CMakeLists.txt
  2. +59
    -0
      tests/ut/cpp/dataset/random_resize_with_bbox_op_test.cc
  3. +54
    -0
      tests/ut/cpp/dataset/resize_with_bbox_op_test.cc
  4. BIN
      tests/ut/data/dataset/imagefolder/ExpectedRandomResizeWithBBox_C0.jpg
  5. BIN
      tests/ut/data/dataset/imagefolder/ExpectedResizeWithBBox_C0.jpg

+ 2
- 0
tests/ut/cpp/dataset/CMakeLists.txt View File

@@ -44,6 +44,7 @@ SET(DE_UT_SRCS
random_horizontal_flip_op_test.cc random_horizontal_flip_op_test.cc
random_horizontal_flip_with_bbox_test.cc random_horizontal_flip_with_bbox_test.cc
random_resize_op_test.cc random_resize_op_test.cc
random_resize_with_bbox_op_test.cc
random_rotation_op_test.cc random_rotation_op_test.cc
random_vertical_flip_op_test.cc random_vertical_flip_op_test.cc
random_vertical_flip_with_bbox_op_test.cc random_vertical_flip_with_bbox_op_test.cc
@@ -53,6 +54,7 @@ SET(DE_UT_SRCS
rescale_op_test.cc rescale_op_test.cc
resize_bilinear_op_test.cc resize_bilinear_op_test.cc
resize_op_test.cc resize_op_test.cc
resize_with_bbox_op_test.cc
schema_test.cc schema_test.cc
shuffle_op_test.cc shuffle_op_test.cc
stand_alone_samplers_test.cc stand_alone_samplers_test.cc


+ 59
- 0
tests/ut/cpp/dataset/random_resize_with_bbox_op_test.cc View File

@@ -0,0 +1,59 @@
/**
* Copyright 2020 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

#include "common/bboxop_common.h"
#include "dataset/kernels/image/random_resize_with_bbox_op.h"
#include "utils/log_adapter.h"

#include "dataset/core/config_manager.h"
#include "dataset/core/global_context.h"

using namespace mindspore::dataset;
using mindspore::LogStream;
using mindspore::ExceptionType::NoExceptionType;
using mindspore::MsLogLevel::INFO;

const bool kSaveExpected = false;
const char kOpName[] = "RandomResizeWithBBox_C";

class MindDataTestRandomResizeWithBBoxOp : public UT::CVOP::BBOXOP::BBoxOpCommon {
protected:
MindDataTestRandomResizeWithBBoxOp() : BBoxOpCommon() {}
};
TEST_F(MindDataTestRandomResizeWithBBoxOp, TestOp) {
MS_LOG(INFO) << "Doing testRandomResizeWithBBox.";
//setting seed here
u_int32_t curr_seed = GlobalContext::config_manager()->seed();
GlobalContext::config_manager()->set_seed(120);
TensorTable results;
std::unique_ptr<RandomResizeWithBBoxOp> op(new RandomResizeWithBBoxOp(500));
for (const auto &tensor_row_ : images_and_annotations_) {
// selected a tensorRow
TensorRow output_row;
Status s = op->Compute(tensor_row_, &output_row);
EXPECT_TRUE(s.IsOk());
results.push_back(output_row);
}
if (kSaveExpected) {
SaveImagesWithAnnotations(FileType::kExpected, std::string(kOpName), results);
}
SaveImagesWithAnnotations(FileType::kActual, std::string(kOpName), results);
if (!kSaveExpected) {
CompareActualAndExpected(std::string(kOpName));
}
GlobalContext::config_manager()->set_seed(curr_seed);
MS_LOG(INFO) << "testRandomResizeWithBBox end.";
}

+ 54
- 0
tests/ut/cpp/dataset/resize_with_bbox_op_test.cc View File

@@ -0,0 +1,54 @@
/**
* Copyright 2020 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

#include "common/bboxop_common.h"
#include "dataset/kernels/image/resize_with_bbox_op.h"
#include "utils/log_adapter.h"

using namespace mindspore::dataset;
using mindspore::LogStream;
using mindspore::ExceptionType::NoExceptionType;
using mindspore::MsLogLevel::INFO;

const bool kSaveExpected = false;
const char kOpName[] = "ResizeWithBBox_C";

class MindDataTestResizeWithBBoxOp : public UT::CVOP::BBOXOP::BBoxOpCommon {
protected:
MindDataTestResizeWithBBoxOp() : BBoxOpCommon() {}
};
TEST_F(MindDataTestResizeWithBBoxOp, TestOp) {
MS_LOG(INFO) << "Doing testResizeWithBBox.";
// resize
TensorTable results;
std::unique_ptr<ResizeWithBBoxOp> op(new ResizeWithBBoxOp(500));
for (const auto &tensor_row_ : images_and_annotations_) {
// selected a tensorRow
TensorRow output_row;
Status s = op->Compute(tensor_row_, &output_row);
EXPECT_TRUE(s.IsOk());
results.push_back(output_row);
}
if (kSaveExpected) {
SaveImagesWithAnnotations(FileType::kExpected, std::string(kOpName), results);
}
SaveImagesWithAnnotations(FileType::kActual, std::string(kOpName), results);
if (!kSaveExpected) {
CompareActualAndExpected(std::string(kOpName));
}

MS_LOG(INFO) << "testResizeWithBBox end.";
}

BIN
tests/ut/data/dataset/imagefolder/ExpectedRandomResizeWithBBox_C0.jpg View File

Before After
Width: 569  |  Height: 500  |  Size: 63 kB

BIN
tests/ut/data/dataset/imagefolder/ExpectedResizeWithBBox_C0.jpg View File

Before After
Width: 569  |  Height: 500  |  Size: 61 kB

Loading…
Cancel
Save