Transforms for video datasets in pytorch - View it on GitHub