fbelderink / flutter_pytorch_mobile

A flutter plugin for pytorch model inference. Supports image models as well as custom models.
https://pub.dev/packages/pytorch_mobile
Other
101 stars 52 forks source link

Custom Mean and Std support #9

Closed elb3k closed 3 years ago

elb3k commented 3 years ago

Added support for custom mean and std for image normalization, with default values from pytorch repo.

elb3k commented 3 years ago

Need to check iOS module.

fbelderink commented 3 years ago

Please request to merge into dev instead of master first :)