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

Hello,

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