diff --git a/mmp/a1/tensors.py b/mmp/a1/tensors.py index 4669751..f96fbac 100644 --- a/mmp/a1/tensors.py +++ b/mmp/a1/tensors.py @@ -2,8 +2,7 @@ import torch def avg_color(img: torch.Tensor): - result = img.mean(dim=(1, 2)).tolist() - return tuple(result) + return img.mean(dim=(1, 2)).tolist() def mask(foreground: torch.Tensor, background: torch.Tensor, mask_tensor: torch.Tensor, threshold: float):