mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
a494a6f6ff
* sync SDK changes according to performance benchmarks * fix end-of-file lint * fix clang-format issue * fix clang-format by adding 'clang-format off' * remove useless casts * remove 'data' argument of 'operator()' * change 'Tensor2Img' to 'TensorToImg' according to spec * correct tensor's name according spec Co-authored-by: lvhan028 <lvhan_028@163.com>
102 lines
2.6 KiB
C++
102 lines
2.6 KiB
C++
// Copyright (c) OpenMMLab. All rights reserved.
|
|
|
|
#include "normalize.h"
|
|
|
|
#include "archive/json_archive.h"
|
|
#include "core/registry.h"
|
|
#include "core/tensor.h"
|
|
|
|
using namespace std;
|
|
|
|
namespace mmdeploy {
|
|
|
|
NormalizeImpl::NormalizeImpl(const Value& args) : TransformImpl(args) {
|
|
if (!args.contains("mean") or !args.contains("std")) {
|
|
ERROR("no 'mean' or 'std' is configured");
|
|
throw std::invalid_argument("no 'mean' or 'std' is configured");
|
|
}
|
|
for (auto& v : args["mean"]) {
|
|
arg_.mean.push_back(v.get<float>());
|
|
}
|
|
for (auto& v : args["std"]) {
|
|
arg_.std.push_back(v.get<float>());
|
|
}
|
|
arg_.to_rgb = args.value("to_rgb", true);
|
|
}
|
|
|
|
/**
|
|
input:
|
|
{
|
|
"ori_img": Mat,
|
|
"img": Tensor,
|
|
"attribute": "",
|
|
"img_shape": [int],
|
|
"ori_shape": [int],
|
|
"img_fields": [int]
|
|
}
|
|
output:
|
|
{
|
|
"img": Tensor,
|
|
"attribute": "",
|
|
"img_shape": [int],
|
|
"ori_shape": [int],
|
|
"img_fields": [string],
|
|
"img_norm_cfg": {
|
|
"mean": [float],
|
|
"std": [float],
|
|
"to_rgb": true
|
|
}
|
|
}
|
|
*/
|
|
|
|
Result<Value> NormalizeImpl::Process(const Value& input) {
|
|
DEBUG("input: {}", to_json(input).dump(2));
|
|
|
|
// copy input data, and update its properties later
|
|
Value output = input;
|
|
|
|
auto img_fields = GetImageFields(input);
|
|
for (auto& key : img_fields) {
|
|
Tensor tensor = input[key].get<Tensor>();
|
|
auto desc = tensor.desc();
|
|
assert(desc.data_type == DataType::kINT8 || desc.data_type == DataType::kFLOAT);
|
|
assert(desc.shape.size() == 4 /*n, h, w, c*/);
|
|
assert(desc.shape[3] == arg_.mean.size());
|
|
|
|
OUTCOME_TRY(output[key], NormalizeImage(tensor));
|
|
|
|
for (auto& v : arg_.mean) {
|
|
output["img_norm_cfg"]["mean"].push_back(v);
|
|
}
|
|
for (auto v : arg_.std) {
|
|
output["img_norm_cfg"]["std"].push_back(v);
|
|
}
|
|
output["img_norm_cfg"]["to_rgb"] = arg_.to_rgb;
|
|
}
|
|
DEBUG("output: {}", to_json(output).dump(2));
|
|
return output;
|
|
}
|
|
|
|
Normalize::Normalize(const Value& args, int version) : Transform(args) {
|
|
auto impl_creator = Registry<NormalizeImpl>::Get().GetCreator(specified_platform_, version);
|
|
if (nullptr == impl_creator) {
|
|
ERROR("'Normalize' is not supported on '{}' platform", specified_platform_);
|
|
throw std::domain_error("'Normalize' is not supported on specified platform");
|
|
}
|
|
impl_ = impl_creator->Create(args);
|
|
}
|
|
|
|
class NormalizeCreator : public Creator<Transform> {
|
|
public:
|
|
const char* GetName() const override { return "Normalize"; }
|
|
int GetVersion() const override { return version_; }
|
|
ReturnType Create(const Value& args) override { return make_unique<Normalize>(args, version_); }
|
|
|
|
private:
|
|
int version_{1};
|
|
};
|
|
|
|
REGISTER_MODULE(Transform, NormalizeCreator);
|
|
|
|
} // namespace mmdeploy
|