Commit c79307b7 authored by leozhang's avatar leozhang Committed by Michael Niedermayer

avfilter/vf_dnn_processing: correct duplicate statement

Signed-off-by: 's avatarleozhang <leozhang@qiyi.com>
Signed-off-by: 's avatarMichael Niedermayer <michael@niedermayer.cc>
parent f6e94225
...@@ -204,7 +204,7 @@ static int config_output(AVFilterLink *outlink) ...@@ -204,7 +204,7 @@ static int config_output(AVFilterLink *outlink)
static int copy_from_frame_to_dnn(DNNData *dnn_data, const AVFrame *in) static int copy_from_frame_to_dnn(DNNData *dnn_data, const AVFrame *in)
{ {
// extend this function to support more formats // extend this function to support more formats
av_assert0(in->format == AV_PIX_FMT_RGB24 || in->format == AV_PIX_FMT_RGB24); av_assert0(in->format == AV_PIX_FMT_RGB24 || in->format == AV_PIX_FMT_BGR24);
if (dnn_data->dt == DNN_FLOAT) { if (dnn_data->dt == DNN_FLOAT) {
float *dnn_input = dnn_data->data; float *dnn_input = dnn_data->data;
...@@ -233,7 +233,7 @@ static int copy_from_frame_to_dnn(DNNData *dnn_data, const AVFrame *in) ...@@ -233,7 +233,7 @@ static int copy_from_frame_to_dnn(DNNData *dnn_data, const AVFrame *in)
static int copy_from_dnn_to_frame(AVFrame *out, const DNNData *dnn_data) static int copy_from_dnn_to_frame(AVFrame *out, const DNNData *dnn_data)
{ {
// extend this function to support more formats // extend this function to support more formats
av_assert0(out->format == AV_PIX_FMT_RGB24 || out->format == AV_PIX_FMT_RGB24); av_assert0(out->format == AV_PIX_FMT_RGB24 || out->format == AV_PIX_FMT_BGR24);
if (dnn_data->dt == DNN_FLOAT) { if (dnn_data->dt == DNN_FLOAT) {
float *dnn_output = dnn_data->data; float *dnn_output = dnn_data->data;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment