how to implement this: torchvision.transforms.functional.perspective

Area Select

react-native-pytorch-core (core package)

Description

Hello! thanks for contributions!

I have a problem while developing my project. I need a function like torchvision.transforms.functional.perspective

Could you add this implementation for torchvision.transforms.functional.perspective? or Can i implement this function? There is no implementation of perspective function in playtorch docs

Another solution that i proceed is making pytorch mobile model for this function. This idea came from @raedle of this issue. but it has a error at react-native app like this:

{"message": "Calling torch.linalg.lstsq on a CPU tensor requires compiling PyTorch with LAPACK. Please use PyTorch built with LAPACK support.

  Debug info for handle(s): debug_handles:{-1}, was not found.

Exception raised from apply_lstsq at ../aten/src/ATen/native/BatchLinearAlgebraKernel.cpp:559 (most recent call first):
(no backtrace available)"}

Should i question to pytorch github about this error?

My perspective model is like this: This model is successful at python code.

import torch, torchvision
from typing import List, Dict
import torchvision.transforms.functional as F

class WrapPerspectiveCrop(torch.nn.Module):
    def __init__(self):
        super().__init__()

    def forward(self, inputs: torch.Tensor, points: List[List[int]]):
        size_points = [[0,0], [inputs.shape[2],0] , [inputs.shape[2],inputs.shape[1]], [0,inputs.shape[1]]]
        inputs = F.perspective(inputs, points, size_points)
        return inputs

    
crop = WrapPerspectiveCrop()
scripted_model = torch.jit.script(crop)
scripted_model.save("wrap_perspective.pt")


import torch
from torch.utils.mobile_optimizer import optimize_for_mobile

optimized_scripted_module=optimize_for_mobile(scripted_model)
optimized_scripted_module._save_for_lite_interpreter("wrap_perspective.ptl")

How can i solve this problem? Many many thanks for anyone help!

[kimishpatel]

I am not entirely sure. Let me get back to you on this. Also is this on ios or android?

[nh9k]

@raedle, I figure out the blocklist was problem. When i removed this blocklist argument optimization_blocklist, the model output works fine as float output on react-native app. Thanks a lot for your help! I need to study more about this blocklist.