All the credit for the dataset generation algorithm goes to SinDongHwan repository 🙏🙏🙏

The pourpose of Traffik package is to transform a bunch of videos (along with their CSVs) generated by TrafficSimulator to a standardized HDF5 dataset used on video summarization tasks (e.g. SumMe or TvSum).

Install pytorch separately and then install the other dependencies:

pip install -r requirements.txt

Install the package locally:

pip install -e .

Generate your dataset.

Usage: python gen.py [OPTIONS]
Options:
 --source TEXT Path of the directory that contains videos and CSVs
 --save TEXT Path where to store the hdf5 dataset file
 --out TEXT Name of the generated HDF5 dataset file.
 --help Show this message and exit.