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

assignment-a1: implements mask function

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

6
mmp/a1/tensors.py
View File

@@ -1,6 +1,4 @@
import torch import torch
from PIL import Image
from torchvision.transforms import ToTensor
def avg_color(img: torch.Tensor): def avg_color(img: torch.Tensor):
@@ -9,9 +7,9 @@ def avg_color(img: torch.Tensor):
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):
raise NotImplementedError() mask = mask_tensor > threshold
return torch.where(mask, foreground, background)
def add_matrix_vector(matrix: torch.Tensor, vector: torch.Tensor): def add_matrix_vector(matrix: torch.Tensor, vector: torch.Tensor):
return matrix.add(vector) return matrix.add(vector)