search

MrinalJain17 / mydia

A simple and efficient wrapper for reading videos as NumPy tensors
https://mrinaljain17.github.io/mydia/
MIT License
25 stars 3 forks source link

Feature: Support for pytorch/theano like tensors - with channels first #5

Closed MrinalJain17 closed 6 years ago

MrinalJain17 commented 6 years ago

PyTorch and theano (and possible other frameworks) have the channels in the first dimension (the zeroth dimension being the batch size).

As of now, the channels are in the last dimension (for frameworks like Tensorflow).

The channels here, refer to the RGB or gray channel.

MrinalJain17 commented 6 years ago

Starting with version 1.0.4, this feature has been implemented - 910b4f41e5d54f816c62312983ce8d77d3be1bfd