OpenSourceJoker/video2x
3
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2024-12-28 06:59:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(realesrgan): fixed incorrect type for ncnn::Mat elemsize

Browse Source 
Signed-off-by: k4yt3x <i@k4yt3x.com>
This commit is contained in:
k4yt3x 2024-10-26 00:00:00 +00:00
parent 50d5cb16d1
commit ebd5e8eb3d
No known key found for this signature in database
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/conversions.cpp
View File

@ -80,7 +80,7 @@ ncnn::Mat avframe_to_ncnn_mat(AVFrame *frame) {
// Allocate a new ncnn::Mat and copy the data // Allocate a new ncnn::Mat and copy the data
int width = converted_frame->width; int width = converted_frame->width;
int height = converted_frame->height; int height = converted_frame->height;
ncnn::Mat ncnn_image = ncnn::Mat(width, height, 3, 3); // BGR has 3 channels ncnn::Mat ncnn_image = ncnn::Mat(width, height, static_cast<size_t>(3), 3);
// Manually copy the pixel data from AVFrame to the new ncnn::Mat // Manually copy the pixel data from AVFrame to the new ncnn::Mat
const uint8_t *src_data = converted_frame->data[0]; const uint8_t *src_data = converted_frame->data[0];

2
src/realesrgan_filter.cpp
View File

@ -113,7 +113,7 @@ int RealesrganFilter::process_frame(AVFrame *in_frame, AVFrame **out_frame) {
// Allocate space for ouptut ncnn::Mat // Allocate space for ouptut ncnn::Mat
int output_width = in_mat.w * realesrgan->scale; int output_width = in_mat.w * realesrgan->scale;
int output_height = in_mat.h * realesrgan->scale; int output_height = in_mat.h * realesrgan->scale;
ncnn::Mat out_mat = ncnn::Mat(output_width, output_height, 3, 3); ncnn::Mat out_mat = ncnn::Mat(output_width, output_height, static_cast<size_t>(3), 3);
ret = realesrgan->process(in_mat, out_mat); ret = realesrgan->process(in_mat, out_mat);
if (ret != 0) { if (ret != 0) {