mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-13 13:50:40 +01:00
286df00f72
This class adds logic for aligning what part of a test video has been encoded from a reference video. It does that by cropping and zooming in on a region of the reference video that most closely matches the test video. A small cropping does not have much impact on human perception, but it has a big impact on PSNR and SSIM calculations. For example, if the test video is cropped with one row in the top and bottom, adjusting for this improves average PSNR from 27.7146 to 29.3357 and average SSIM from 0.934891 to 0.95318 in an example test video. TBR=phoglund Bug: webrtc:9642 Change-Id: I02cfe0e2261fb58df8cdb1e15ba93285e3dc4538 Reviewed-on: https://webrtc-review.googlesource.com/c/99480 Commit-Queue: Magnus Jedvert <magjed@webrtc.org> Reviewed-by: Patrik Höglund <phoglund@google.com> Reviewed-by: Sami Kalliomäki <sakal@webrtc.org> Cr-Commit-Position: refs/heads/master@{#25755}
57 lines
2.2 KiB
C++
57 lines
2.2 KiB
C++
/*
|
|
* Copyright (c) 2018 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#ifndef RTC_TOOLS_FRAME_ANALYZER_VIDEO_GEOMETRY_ALIGNER_H_
|
|
#define RTC_TOOLS_FRAME_ANALYZER_VIDEO_GEOMETRY_ALIGNER_H_
|
|
|
|
#include "api/video/video_frame_buffer.h"
|
|
#include "rtc_tools/video_file_reader.h"
|
|
|
|
namespace webrtc {
|
|
namespace test {
|
|
|
|
struct CropRegion {
|
|
// Each value represents how much to crop from each side. Left is where x=0,
|
|
// and top is where y=0. All values equal to zero represents no cropping.
|
|
int left = 0;
|
|
int right = 0;
|
|
int top = 0;
|
|
int bottom = 0;
|
|
};
|
|
|
|
// Crops and zooms in on the cropped region so that the returned frame has the
|
|
// same resolution as the input frame.
|
|
rtc::scoped_refptr<I420BufferInterface> CropAndZoom(
|
|
const CropRegion& crop_region,
|
|
const rtc::scoped_refptr<I420BufferInterface>& frame);
|
|
|
|
// Calculate the optimal cropping region on the reference frame to maximize SSIM
|
|
// to the test frame.
|
|
CropRegion CalculateCropRegion(
|
|
const rtc::scoped_refptr<I420BufferInterface>& reference_frame,
|
|
const rtc::scoped_refptr<I420BufferInterface>& test_frame);
|
|
|
|
// Returns a cropped and zoomed version of the reference frame that matches up
|
|
// to the test frame. This is a simple helper function on top of
|
|
// CalculateCropRegion() and CropAndZoom().
|
|
rtc::scoped_refptr<I420BufferInterface> AdjustCropping(
|
|
const rtc::scoped_refptr<I420BufferInterface>& reference_frame,
|
|
const rtc::scoped_refptr<I420BufferInterface>& test_frame);
|
|
|
|
// Returns a cropped and zoomed version of the reference video that matches up
|
|
// to the test video. Frames are individually adjusted for cropping.
|
|
rtc::scoped_refptr<Video> AdjustCropping(
|
|
const rtc::scoped_refptr<Video>& reference_video,
|
|
const rtc::scoped_refptr<Video>& test_video);
|
|
|
|
} // namespace test
|
|
} // namespace webrtc
|
|
|
|
#endif // RTC_TOOLS_FRAME_ANALYZER_VIDEO_GEOMETRY_ALIGNER_H_
|