Home
last modified time | relevance | path

Searched refs:convert_mode (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/ir/vision/
Dconvert_color_ir.cc31 ConvertColorOperation::ConvertColorOperation(ConvertMode convert_mode) : convert_mode_(convert_mode… in ConvertColorOperation() argument
60 ConvertMode convert_mode = static_cast<ConvertMode>(op_params["convert_mode"]); in from_json() local
61 *operation = std::make_shared<vision::ConvertColorOperation>(convert_mode); in from_json()
Dconvert_color_ir.h40 explicit ConvertColorOperation(ConvertMode convert_mode);
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/
Dconvert_color_op.cc28 ConvertColorOp::ConvertColorOp(ConvertMode convert_mode) : convert_mode_(convert_mode) {} in ConvertColorOp() argument
Dconvert_color_op.h32 explicit ConvertColorOp(ConvertMode convert_mode);
Dimage_utils.cc67 Status GetConvertShape(ConvertMode convert_mode, const std::shared_ptr<CVTensor> &input_cv, in GetConvertShape() argument
78 …if (std::find(three_channels.begin(), three_channels.end(), convert_mode) != three_channels.end())… in GetConvertShape()
80 …} else if (std::find(four_channels.begin(), four_channels.end(), convert_mode) != four_channels.en… in GetConvertShape()
82 …} else if (std::find(one_channels.begin(), one_channels.end(), convert_mode) != one_channels.end()… in GetConvertShape()
485 …(const std::shared_ptr<Tensor> &input, std::shared_ptr<Tensor> *output, ConvertMode convert_mode) { in ConvertColor() argument
500 RETURN_IF_NOT_OK(GetConvertShape(convert_mode, input_cv, &node)); in ConvertColor()
509 cv::cvtColor(input_cv->mat(), output_cv->mat(), static_cast<int>(convert_mode)); in ConvertColor()
Dimage_utils.h141 …r(const std::shared_ptr<Tensor> &input, std::shared_ptr<Tensor> *output, ConvertMode convert_mode);
/third_party/mindspore/mindspore/dataset/vision/
Dc_transforms.py315 def __init__(self, convert_mode): argument
316 self.convert_mode = convert_mode
319 return cde.ConvertColorOperation(DE_C_CONVERTCOLOR_MODE[self.convert_mode])
Dvalidators.py974 [convert_mode], _ = parse_user_args(method, *args, **kwargs)
975 if convert_mode is not None:
976 type_check(convert_mode, (ConvertMode,), "convert_mode")
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Dvision.cc207 explicit Data(ConvertMode convert_mode) : convert_mode_(convert_mode) {} in Data()
211 ConvertColor::ConvertColor(ConvertMode convert_mode) : data_(std::make_shared<Data>(convert_mode)) … in ConvertColor() argument
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/kernels/ir/image/
Dbindings.cc121 .def(py::init([](ConvertMode convert_mode) { in __anon574f76b40902() argument
122 auto convert = std::make_shared<vision::ConvertColorOperation>(convert_mode); in __anon574f76b40902()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/
Dvision.h121 explicit ConvertColor(ConvertMode convert_mode);