homedeck / deep-edit Goto Github PK

Unknown type constructor Union:
  File "/home/ubuntu/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/plasma/filters/spectral.py", line 52
def shadows (input: Tensor, weight: Union[float, Tensor], tonal_range: float=1.) -> Tensor:
                                    ~~~~~~~~~~~~~~~~~~~ <--- HERE
    """
    Apply shadow attentuation to an image.
'shadows' is being compiled since it was called from 'DeepEdit.edit'
  File "/home/ubuntu/Deep-Edit/model.py", line 71
        batch, _, height, width = input.shape
        # Linear
        input = shadows(input, 0.75)
        ~~~~~~~~~~~~~~~~~~~~~~~~~~~ <--- HERE
        input = highlights(input, -0.8)
        input = contrast(input, coefficients[:,0].view(-1, 1, 1, 1))
'DeepEdit.edit' is being compiled since it was called from 'DeepEdit.forward'
  File "/home/ubuntu/Deep-Edit/model.py", line 41
    def forward (self, input: Tensor) -> Tensor:
        weights = self.coefficients(input)
        result = self.edit(input, weights)
        ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ <--- HERE
        return result