iamgmujtaba / ltc-sum

Implementation of LTC-SUM: Lightweight Client-driven Personalized Video Summarization Framework Using 2D CNN

LTC-SUM: Lightweight Client-driven Personalized Video Summarization Framework Using 2D CNN

This repository contains the original implementation of the paper LTC-SUM: Lightweight Client-driven Personalized Video Summarization Framework Using 2D CNN, published in IEEE Access 2022.

Abstract

This paper proposes a novel lightweight thumbnail container-based summarization (LTC-SUM) framework for full feature-length videos. This framework generates a personalized keyshot summary for concurrent users by using the computational resource of the end-user device. State-of-the-art methods that acquire and process entire video data to generate video summaries are highly computationally intensive. In this regard, the proposed LTC-SUM method uses lightweight thumbnails to handle the complex process of detecting events. This significantly reduces computational complexity and improves communication and storage efficiency by resolving computational and privacy bottlenecks in resource-constrained end-user devices. These improvements were achieved by designing a lightweight 2D CNN model to extract features from thumbnails, which helped select and retrieve only a handful of specific segments. Extensive quantitative experiments on a set of full 18 feature-length videos (approximately 32.9 h in duration) showed that the proposed method is significantly computationally efficient than state-of-the-art methods on the same end-user device configurations. Joint qualitative assessments of the results of 56 participants showed that participants gave higher ratings to the summaries generated using the proposed method. To the best of our knowledge, this is the first attempt in designing a fully client-driven personalized keyshot video summarization framework using thumbnail containers for feature-length videos.

Prerequisite

• Linux or macOS
• Python 3.6
• CPU or NVIDIA GPU + CUDA CuDNN

Getting Started

Installation

• Clone this repo:

git clone https://github.com/iamgmujtaba/LTC-SUM
cd LTC-SUM

• To create conda environment and install cuda toolkit, run the following command:

conda create -n ltcsum cudatoolkit=10.0 cudnn=7.6.0 python=3.6 -y
conda activate ltcsum

• Install TensorFlow and Keras and other dependencies
  • For pip users, please type the command

pip install -r requirements.txt

Preparing the data

1. Create train and test folders

cd data && mkdir train && mkdir test

2. Download the dataset from UCF into the data folder:

wget wget https://www.crcv.ucf.edu/data/UCF101/UCF101.rar --no-check-certificate

3. Extract UCF101.rar file in the data folder

unrar e UCF101.rar

4. Run the scripts in the data folder to move the videos to the appropriate folders

python 1_move_files_ucf101.py 

5. Run the scripts in the data folder to extract video frames in the train/test folders and make the CSV file. The CSV file will be used in the rest of the code references

python 2_extract_files_ucf101.py

• Note: You need FFmpeg installed to extract frames from videos.

Train and evaluate

To train the model, run the following command.

python train.py --dataset_path /path/to/UCF101 --model_name efficientNet --batch_size 32 --epochs 100 --learning_rate 0.001 --num_classes 101 --save_model_path /path/to/save/model

Check config.py for the list of all the parameters.

• In order to evaluate the proposed method, you have to configure hls-server.
• Use vid2tc to generate thumbnail contaienrs from videos. For more inforamtion, please refer to the paper.
• Download the pretrained model from google drive.
• Place the pretrained model in the output folder.
• Run the following command to test the proposed method.

python test.py --genre wt 

Experimental Results

generated_summary.mp4

Citation

If you use this code for your research, please cite our paper.

@ARTICLE{mujtaba2022ltc,
  title={LTC-SUM: Lightweight Client-Driven Personalized Video Summarization Framework Using 2D CNN}, 
  author={Mujtaba, Ghulam and Malik, Adeel and Ryu, Eun-Seok},
  journal={IEEE Access}, 
  year={2022},
  volume={10},
  number={},
  pages={103041-103055},
  doi={10.1109/ACCESS.2022.3209275}}

The following papers are also related to this research, please cite the articles if you use the code.

@inproceedings{mujtabasigmap2022,
    title={Client-driven Lightweight Method to Generate Artistic Media for Feature-length Sports Videos},
    author={Mujtaba, Ghulam and Choi, Jaehyuk and Ryu, Eun-Seok},
    booktitle={19th International Conference on Signal Processing and Multimedia Applications (SIGMAP)},
    pages={102-111},
    year={2022},
    address = {Lisbon, Portugal},
    month = {}}

@article{mujtaba2020client,
  title={Client-driven personalized trailer framework using thumbnail containers},
  author={Mujtaba, Ghulam and Ryu, Eun-Seok},
  journal={IEEE Access},
  volume={8},
  pages={60417--60427},
  year={2020},
  publisher={IEEE}
}

@article{mujtabamtap2021,
  title={Client-driven animated GIF generation framework using an acoustic feature},
  author={Mujtaba, Ghulam and Lee, Sangsoon and Kim, Jaehyoun and Ryu, Eun-Seok},
  journal={Multimedia Tools and Applications},
  year={2021},
  publisher={Springer}}

@inproceedings{mujtaba2021human,
  title={Human character-oriented animated gif generation framework},
  author={Mujtaba, Ghulam and Ryu, Eun-Seok},
  booktitle={2021 Mohammad Ali Jinnah University International Conference on Computing (MAJICC)},
  pages={1--6},
  year={2021},
  organization={IEEE}
}