simonfranken/mmp_wise2526_franksim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

assignment-a1: fix

Browse Source
This commit is contained in:
franksim
2025-10-16 14:15:40 +00:00
parent 9372abd0a3
commit caad6dc779
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
mmp/a1/tensors.py
View File

@@ -2,8 +2,7 @@ import torch
def avg_color(img: torch.Tensor): def avg_color(img: torch.Tensor):
result = img.mean(dim=(1, 2)).tolist() return img.mean(dim=(1, 2)).tolist()
return tuple(result)
def mask(foreground: torch.Tensor, background: torch.Tensor, mask_tensor: torch.Tensor, threshold: float): def mask(foreground: torch.Tensor, background: torch.Tensor, mask_tensor: torch.Tensor, threshold: float):