Skip to content

Latest commit

 

History

History
30 lines (20 loc) · 1.01 KB

README.md

File metadata and controls

30 lines (20 loc) · 1.01 KB

traffik

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.