Function blob

models/utils.py:40–50

(im: ndarray, return_seg: bool = False)

def blob(im: ndarray, return_seg: bool = False) -> ndarray | tuple:
    seg = None
    if return_seg:
        seg = im.astype(np.float32) / 255
    im = im.transpose([2, 0, 1])
    im = im[np.newaxis, ...]
    im = np.ascontiguousarray(im).astype(np.float32) / 255
    if return_seg:
        return im, seg
    else:
        return im


def sigmoid(x: ndarray) -> ndarray: