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