[Torchvision] [Android] Torchvision.ops when will they be available on Mobile


Let me introduce a problem. Last month two significant features for mobile deployment was added to pytorch=1.4. I am talking about torchvision operations torchvision::nms and torchvision::roi_align that are now convertible from eager python code via torchscript to .pt model.

The most recent builds of pytorch for mobile don’t contain this features. Is there any estimation when wiil tey be available to use on phones?

We’re looking into this. Probably late this year or early next year.

1 Like

Thank you so much for the update. Would nms be planned to support on Android anytime soon?


Hello, When nms will be supported in Android?


Any further progress?

Any update on the NMS for Pytorch Mobile (Android) ?