Li Zhang 28beb9a481
[Enhancement] Adjust csrc structure (#594)
* adjust csrc structure

* adjust csrc includes

* fix lint

* fix lint

* fix MSVC build

* remove useless code
2022-06-14 13:46:01 +08:00

20 lines
604 B
C++

// Copyright (c) OpenMMLab. All rights reserved.
#ifndef MMDEPLOY_TEST_TRANSFORM_UTILS_H
#define MMDEPLOY_TEST_TRANSFORM_UTILS_H
#include "mmdeploy/core/tensor.h"
#include "mmdeploy/core/value.h"
#include "mmdeploy/preprocess/transform/transform.h"
namespace mmdeploy::test {
std::unique_ptr<Transform> CreateTransform(const Value& cfg, Device device, Stream stream);
std::vector<int64_t> Shape(const Value& value, const std::string& shape_key);
std::vector<float> ImageNormCfg(const Value& value, const std::string& key);
} // namespace mmdeploy::test
#endif // MMDEPLOY_TEST_TRANSFORM_UTILS_H