4 lines
88 B
Python
4 lines
88 B
Python
from torch import Tensor
|
|
def image_to_mask(image:Tensor):
|
|
return image[..., 2]
|
|
|