mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
ea54f3b2fd
* keep DefaultFormatBundle * add DefaultFormatBundle * add condition * resolve comments * remove useless * add override
50 lines
1.1 KiB
C++
50 lines
1.1 KiB
C++
// Copyright (c) OpenMMLab. All rights reserved.
|
|
|
|
#ifndef MMDEPLOY_CROP_H
|
|
#define MMDEPLOY_CROP_H
|
|
|
|
#include <array>
|
|
|
|
#include "core/tensor.h"
|
|
#include "transform.h"
|
|
|
|
namespace mmdeploy {
|
|
|
|
class MMDEPLOY_API CenterCropImpl : public TransformImpl {
|
|
public:
|
|
explicit CenterCropImpl(const Value& args);
|
|
~CenterCropImpl() override = default;
|
|
|
|
Result<Value> Process(const Value& input) override;
|
|
|
|
protected:
|
|
virtual Result<Tensor> CropImage(const Tensor& tensor, int top, int left, int bottom,
|
|
int right) = 0;
|
|
|
|
protected:
|
|
struct center_crop_arg_t {
|
|
std::array<int, 2> crop_size;
|
|
};
|
|
using ArgType = struct center_crop_arg_t;
|
|
|
|
protected:
|
|
ArgType arg_;
|
|
};
|
|
|
|
class MMDEPLOY_API CenterCrop : public Transform {
|
|
public:
|
|
explicit CenterCrop(const Value& args, int version = 0);
|
|
~CenterCrop() override = default;
|
|
|
|
Result<Value> Process(const Value& input) override { return impl_->Process(input); }
|
|
|
|
protected:
|
|
std::unique_ptr<CenterCropImpl> impl_;
|
|
};
|
|
|
|
MMDEPLOY_DECLARE_REGISTRY(CenterCropImpl);
|
|
|
|
} // namespace mmdeploy
|
|
|
|
#endif // MMDEPLOY_CROP_H
|