Blame view

3rdparty/opencv-4.5.4/modules/videoio/perf/perf_output.cpp 1.5 KB
f4334277   Hu Chunming   提交3rdparty
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
  // This file is part of OpenCV project.
  // It is subject to the license terms in the LICENSE file found in the top-level directory
  // of this distribution and at http://opencv.org/license.html
  #include "perf_precomp.hpp"
  
  namespace opencv_test
  {
  using namespace perf;
  
  typedef tuple<std::string, bool> VideoWriter_Writing_t;
  typedef perf::TestBaseWithParam<VideoWriter_Writing_t> VideoWriter_Writing;
  
  const string image_files[] = {
      "python/images/QCIF_00.bmp",
      "python/images/QCIF_01.bmp",
      "python/images/QCIF_02.bmp",
      "python/images/QCIF_03.bmp",
      "python/images/QCIF_04.bmp",
      "python/images/QCIF_05.bmp"
  };
  
  PERF_TEST_P(VideoWriter_Writing, WriteFrame,
              testing::Combine(
                  testing::ValuesIn(image_files),
                  testing::Bool()))
  {
    const string filename = getDataPath(get<0>(GetParam()));
    const bool isColor = get<1>(GetParam());
    Mat image = imread(filename, isColor ? IMREAD_COLOR : IMREAD_GRAYSCALE );
  #if defined(HAVE_MSMF) && !defined(HAVE_FFMPEG)
    const string outfile = cv::tempfile(".wmv");
    const int fourcc = VideoWriter::fourcc('W', 'M', 'V', '3');
  #else
    const string outfile = cv::tempfile(".avi");
    const int fourcc = VideoWriter::fourcc('X', 'V', 'I', 'D');
  #endif
  
    VideoWriter writer(outfile, fourcc, 25, cv::Size(image.cols, image.rows), isColor);
    if (!writer.isOpened())
        throw SkipTestException("Video file can not be opened");
  
    TEST_CYCLE_N(100) { writer << image; }
    SANITY_CHECK_NOTHING();
    remove(outfile.c_str());
  }
  
  } // namespace