mirror of
https://github.com/k4yt3x/video2x.git
synced 2025-01-15 12:18:52 +00:00
24 lines
515 B
C++
24 lines
515 B
C++
#pragma once
|
|
|
|
extern "C" {
|
|
#include <libavutil/frame.h>
|
|
#include <libswscale/swscale.h>
|
|
}
|
|
|
|
#include <mat.h>
|
|
|
|
namespace video2x {
|
|
namespace conversions {
|
|
|
|
// Convert AVFrame to another pixel format
|
|
AVFrame* convert_avframe_pix_fmt(AVFrame* src_frame, AVPixelFormat pix_fmt);
|
|
|
|
// Convert AVFrame to ncnn::Mat
|
|
ncnn::Mat avframe_to_ncnn_mat(AVFrame* frame);
|
|
|
|
// Convert ncnn::Mat to AVFrame
|
|
AVFrame* ncnn_mat_to_avframe(const ncnn::Mat& mat, AVPixelFormat pix_fmt);
|
|
|
|
} // namespace conversions
|
|
} // namespace video2x
|