static int copy_from_frame_to_dnn(DNNData *dnn_data, const AVFrame *in)
{
// 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) {
float *dnn_input = dnn_data->data;
static int copy_from_dnn_to_frame(AVFrame *out, const DNNData *dnn_data)
{
// 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) {
float *dnn_output = dnn_data->data;