File size: 148 Bytes
1ebfbbf | 1 2 3 4 5 6 7 | import numpy as np
def postprocess(mask):
if isinstance(mask, dict):
mask = mask["mask"]
return np.asarray(mask, dtype=np.float32)
|
1ebfbbf | 1 2 3 4 5 6 7 | import numpy as np
def postprocess(mask):
if isinstance(mask, dict):
mask = mask["mask"]
return np.asarray(mask, dtype=np.float32)
|