Add Imagenet PreProcessor

mlpack / models

models built with mlpack

https://models.mlpack.org/docs

BSD 3-Clause "New" or "Revised" License

35 stars 41 forks source link

Add Imagenet PreProcessor #27

Closed kartikdutt18 closed 4 years ago

kartikdutt18 commented 4 years ago

Hey everyone, This PR adds the functional equivalent of PyTorch's to tensor function. Since we have transferred weights from PyTorch the input Feature matrix also need to match with PyTorch so this PR adds the same. Docs.