Forum | Documentation | Website | Blog

Skip to content
Snippets Groups Projects
commercial_detection_and_replacement.rst 25.67 KiB

Enhanced Media Experience with AI-Powered Commercial Detection and Replacement

Introduction

Leveraging the capabilities of BeagleBoard’s powerful processing units, the project will focus on creating a real-time, efficient solution that enhances media consumption experiences by seamlessly integrating custom audio streams during commercial breaks.

Summary links

Status

This project is currently just a proposal.

Proposal

About

  • Resume - Find my resume here
  • Forum: :fab:`discourse` u/aryan_nanda
  • OpenBeagle: :fab:`gitlab` aryan_nanda
  • Github: :fab:`github` AryanNanda17
  • School: :fas:`school` Veermata Jijabai Technological Institute (VJTI)
  • Country: :fas:`flag` India
  • Primary language: :fas:`language` English, Hindi
  • Typical work hours: :fas:`clock` 9AM-5PM Indian Standard Time
  • Previous GSoC participation: :fab:`google` This would be my first time participating in GSOC

About the Project

Project name: Enhanced Media Experience with AI-Powered Commercial Detection and Replacement

Description

I propose developing a GStreamer Plugin capable of processing video inputs based on their classification. The plugin will identify commercials and either replace them with alternative content or obscure them, while also substituting the audio with predefined streams. This enhancement aims to improve the media consumption experience by eliminating unnecessary interruptions. I intend to explore various video classification models to achieve accurate detection and utilize TensorFlow Lite to leverage the native accelerators of BeagleBone AI-64 for high-performance, real-time inferencing with minimal latency. I believe real-time high-performance would be the most critical thing for this project and I intend on testing a few different ways to see which one works best.

Goals and Objectives

The goal of this project is to detect and replace commercials in video streams on BeagleBoard hardware using a GStreamer pipeline which includes a model that accurately detects commercials with minimal latency. Comparison of different model accuracy can be done by doing some manual analysis and trying different video classification models and to finally use the best performing option to be included in the GStreamer pipeline for inferencing of real-time videos. This would be the result presented at the end of the project timeline. For phase 1 evaluation, the goal is to build a training dataset, preprocess it and fine-tune and train a Video Classification model to identify commercials segments in a video accurately. For phase 2 evaluation, the goal is to use the the best model identified in phase 1 for commercial detection and build a GStreamer pipeline which would do video processing based on commercial segments classification and using native accelerators present in BeagleBone-ai-64 for high-performance.

In order to accomplish this project the following objectives need to be met.

  1. Phase 1:-
    • Develop a dataset of videos and corresponding labels indicating the presence of commercials in specific segments.
    • Preprocess the dataset to ensure it's suitable for input into deep learning models. Moreover divide the datset into train, validation and test set.
    • Fine-tune various deep learning models and train them on the prepared dataset to identify the most accurate one for commercial detection in videos.
    • Save all trained models to local disk and perform real-time inference using OpenCV to determine the model that yields the best results with high-performance.
  2. Phase 2:-
    • Based on all the options tried in Phase 1, decide on the final model to be used in the GStreamer pipeline.
    • Compiling the model and generating artifacts so that we can use it in TFLite Runtime.
    • Building a GStreamer pipeline that would take real-time input of media and would identify the commercial segments in it.
    • If the commercial segment is identified the GStreamer pipeline would either replace them with alternative content or obscure them, while also substituting the audio with predefined streams.
    • Enhancing the Real-time performance using native hardware Accelerators present in BeagleBone-Ai-64.

Methods

In this section, I will individually specify the training dataset, model, GStreamer Pipeline etc. methods that I plan on using in greater details.

Building training Dataset

To train the model effectively, we need a dataset with accurate labels. Since a suitable commercial video dataset isn't readily available, I'll create one. This dataset will consist of two classes: commercial and non-commercial. To build this dataset, I'll refer to the Youtube-8M dataset, which includes videos categorized as TV advertisements. I'll download these videos and organize them into two folders: commercial and non-commercial. However, since the Youtube-8M dataset provides encoded feature vectors instead of the actual videos, direct usage would result in significant latency. Therefore, I'll use it as a reference and download the videos labeled by it as advertisements to build our dataset. After the dataset is ready I will preprocess it to ensure it's suitable for input into deep learning models. Moreover I'll divide the datset into train, validation and test set.

Video Classification models

MoViNets is a good model for our task as it can operate on streaming videos for online inference. The main reason behind trying out MoViNets first is becaue it does quick and continuous analysis of incoming video streams. MoViNet utilizes NAS(Neural Architecture Search) to balance accuracy and efficiency, incorporates stream buffers for constant memory usage, and improves accuracy via temporal ensembles. The MoViNet architecture uses 3D convolutions that are "causal". Causal convolution ensures that the output at time t is computed using only inputs up to time t. This allows for efficient streaming. This make MoViNets a perfect choice for our case. So, I will fine tune the MoViNets model and will train it on the commercial detection dataset.

Stream buffer in MoViNets Standard Convolution Vs Causal Convolution

If MoViNet does not perform well than we can use other models like ResNet-50+LSTMs. Since a video is just a series of frames, a naive video classification method would be pass each frame from a video file through a CNN, classify each frame individually and independently of each other, choose the label with the largest corresponding probability, label the frame, and assign the most assigned image label to the video. To solve the problem of "prediction flickering", where the label for the video changes rapidly when scenes get labeled differently. I will use rolling prediction averaging to reduce “flickering” in results.

The Conv+LSTMs model will perform well as it considers both the spatial and temporal features of videos just like a Conv3D model. The only reason it is not my first choice is because MoViNets are considered to be better for real-time performance.

Conv3D+LSTMs

Optional method with Video Vision Transformers

This is a pure Transformer based model which extracts spatio-temporal tokens from the input video, which are then encoded by a series of transformer layers. I have kept this as an optional method because our problem is of binary classification(either Commercial or Non-Commercial), so using such a complex model for this small problem may not be as efficient as other models.

Choosing the Best Performing model

After training the models, I'll assess their performance using evaluation metrics and conduct real-time inference on a sample video containing both commercial and non-commercial segments. I'll select the model with the highest accuracy and integrate it into the GStreamer pipeline for further processing.

Model Execution on BeagleBone AI-64

The BeagleBone AI-64 Linux for Edge AI supports importing pre-trained custom models to run inference on target. Moreover, Edge AI BeagleBone AI-64 images have TensorFlow Lite already installed with acceleration enabled. The Debian-based SDK makes use of pre-compiled DNN (Deep Neural Network) models and performs inference using various OSRT (open source runtime) such as TFLite runtime, ONNX runtime etc.

In order to infer a DNN, SDK expects the DNN and associated artifacts in the below directory structure.

project_root

├── param.yaml

├── artifacts
│   ├── 264_tidl_io_1.bin
│   ├── 264_tidl_net.bin
│   ├── 264_tidl_net.bin.layer_info.txt
│   ├── 264_tidl_net.bin_netLog.txt
│   ├── 264_tidl_net.bin.svg
│   ├── allowedNode.txt
│   └── runtimes_visualization.svg

└── model
    └── ssd_mobilenet_v2_300_float.tflite
  1. model: This directory contains the DNN being targeted to infer.
  2. artifacts: This directory contains the artifacts generated after the compilation of DNN for SDK.
  3. param.yaml: A configuration file in yaml format to provide basic information about DNN, and associated pre and post processing parameters.

Therefore, after choosing the model to be used in GStreamer pipeline, I will generate the artifacts directory by following these instructions.

TFLite Runtime

GStreamer Pipeline

The data flow in the GStreamer pipeline at a high level can be split into 3-parts:-

  1. Input Pipeline - Grabs a frame from the input source.
  2. Output Pipeline - Sends the output to the display.
  3. Compute Pipeline - Performs pre-processing, inference and post-processing.

I will create a GStreamer Pipeline that will receive an input of a video and it will grab it frame by frame. The frame will be split into two paths. The “analytics” path resizes the input maintaining the aspect ratio and crops the input to match the resolution required to run the deep learning network. The “visualization” path is provided to the post-processing module which overlays the detected classes. If a commercial video is detected, we apply blurring to the video frames and replace the audio. If a non-commercial video is detected, proceed with the normal visualization process without blurring or replacing the audio. Post-processed output is given to HW mosaic plugin which positions and resizes the output window on an empty background before sending to display.

The following GStreamer input and output pipeline describes how I will build the GStreamer pipeline:-

  • GStreamer input pipeline:
v4l2src device=/dev/video0 ! \
video/x-raw,format=NV12,width=1920,height=1080 ! \
tiovxmultiscaler name=split_01 ! \
split_01. ! queue ! video/x-raw, width=320, height=320 ! \
tiovxdlpreproc data-type=10 channel-order=1 mean-0=128.000000 mean-1=128.000000 mean-2=128.000000 scale-0=0.007812 scale-1=0.007812 scale-2=0.007812 tensor-format=rgb out-pool-size=4 ! \
application/x-tensor-tiovx ! appsink name=pre_0 max-buffers=2 drop=true \
split_01. ! queue ! video/x-raw, width=1280, height=720 ! \
tiovxdlcolorconvert target=1 out-pool-size=4 ! \
video/x-raw, format=RGB ! appsink name=sen_0 max-buffers=2 drop=true
  • GStreamer output pipeline:
appsrc format=GST_FORMAT_TIME is-live=true block=true do-timestamp=true name=post_0 ! \
    tiovxdlcolorconvert ! video/x-raw,format=NV12,width=1280,height=720 ! \
    queue ! mosaic_0.sink_0

appsrc format=GST_FORMAT_TIME block=true num-buffers=1 name=background_0 ! \
    tiovxdlcolorconvert ! video/x-raw,format=NV12,width=1920,height=1080 ! \
    queue ! mosaic_0.background

tiovxmosaic name=mosaic_0 \
    sink_0::startx=320  sink_0::starty=180  sink_0::width=1280  sink_0::height=720 \
    ! video/x-raw,format=NV12,width=1920,height=1080 ! \
    kmssink sync=false driver-name=tidss
GStreamer Pipeline Project Workflow

Software

Hardware

Timeline

Timeline summary

Date Activity
April 3 - May 1 Understanding GStreamer pipeline and TFLite runtime of BeagleBone-Ai-64.
May 1 - May 10 Discussing implementation ideas with mentors.
May 10 - May 31 Focus on college exams.
June 1 - June 3 Start coding and introductory video
June 3 :ref:`Release introductory video and complete milestone #1<Milestone1>`
June 10 :ref:`Complete milestone #2<Milestone2>`
June 17 :ref:`Complete milestone #3<Milestone3>`
June 24 :ref:`Complete milestone #4<Milestone4>`
July 1 :ref:`Complete milestone #5<Milestone5>`
July 8 Submit midterm evaluations
July 15 :ref:`Complete milestone #6<Milestone6>`
July 22 :ref:`Complete milestone #7<Milestone7>`
July 29 :ref:`Complete milestone #8<Milestone8>`
August 5 :ref:`Complete milestone #9<Milestone9>`
August 12 :ref:`Complete milestone #10<Milestone10>`
August 19 Submit final project video, submit final work to GSoC site and complete final mentor evaluation

Timeline detailed

Community Bonding Period (May 1st - May 10th)

  • Discuss implementation idea with mentors.
  • Discuss the Scope of the project.

Milestone #1, Introductory YouTube video (June 3rd)

  • Making an Introductory Video.
  • Start working on dataset.

Milestone #2 (June 10th)

  • Develop a dataset of videos and corresponding labels indicating the presence of commercials in specific segments.
  • Preprocess the dataset to ensure it’s suitable for input into deep learning models. Moreover divide the datset into train, validation and test set.

Milestone #3 (June 17th)

  • Fine-tune various deep learning models and train them on the prepared dataset to identify the most accurate one for commercial detection in videos.

Milestone #4 (June 24th)

  • Continue working on model and fine-tuning it to give the best results.

Milestone #5 (July 1st)

  • Save all trained models to local disk and perform real-time inference using OpenCV to determine the model that yields the best results with high-performance.

Submit midterm evaluations (July 8th)

  • Completing remaining phase 1 tasks if any.

Important

July 12 - 18:00 UTC: Midterm evaluation deadline (standard coding period)

Milestone #6 (July 15th)

  • Based on all the options tried in Phase 1, decide on the final model to be used in the GStreamer pipeline.
  • Compiling the model and generating artifacts so that we can use it in TFLite Runtime.

Milestone #7 (July 22nd)

  • Building a GStreamer pipeline that would take real-time input of media and would identify the commercial segments in it.

Milestone #8 (July 29th)

  • Continue working on the GStreamer pipeline.
  • If the commercial segment is identified the GStreamer pipeline would either replace them with alternative content or obscure them, while also substituting the audio with predefined streams.

Milestone #9 (Aug 5th)

  • Enhancing the Real-time performance using native hardware Accelerators present in BeagleBone-Ai-64.

Milestone #10 (Aug 12th)

  • Continue working on optimizing the real-time output for best performance.

Final YouTube video (Aug 19th)

  • Submit final project video, submit final work to GSoC site and complete final mentor evaluation.