mv_machine_learning/training/src/data_augment_flip.cpp

   1 /**
   2  * Copyright (c) 2022 Samsung Electronics Co., Ltd All Rights Reserved
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include "data_augment_flip.h"
  18
  19 using namespace std;
  20
  21 DataAugmentFlip::DataAugmentFlip() : DataAugment()
  22 {}
  23
  24 DataAugmentFlip::~DataAugmentFlip()
  25 {}
  26
  27 void DataAugmentFlip::Preprocess(vector<float> &in_vec, vector<float> &out_vec, int width, int height)
  28 {
  29         cv::Mat cvSrc = cv::Mat(cv::Size(width, height), CV_32FC3, in_vec.data()).clone();
  30
  31         cv::Mat cvFlip;
  32
  33         cv::flip(cvSrc, cvFlip, 1);
  34
  35         out_vec.assign((float *) cvFlip.data, (float *) cvFlip.data + cvFlip.total() * cvFlip.channels());
  36 }