search

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

Torch 1.8.0 update and torchvision ops support #8

Closed elb3k closed 3 years ago

elb3k commented 3 years ago

Updated torch 1.6.0 to torch 1.8.0, and added torchvision_ops support for android at least (Don't have mac device). And there was a bug here: https://github.com/fynnmaarten/flutter_pytorch_mobile/blob/3f0dcbcb560668d36828a8d7d18c21549e46c34a/android/src/main/java/io/fynn/pytorch_mobile/PyTorchMobilePlugin.java#L135-L136 where the order of data and shape was mistaken.

And I have added torchvision_ops support for android here, it is a very useful feature for object detection models.