util.py 文件源码

def get_original_size(mask, max_size=(640,640)):
    row = None
    col = None
    for i in range(max_size[0]-1, -1, -1):
        if mask[i,0,0] == 1:
            row = i + 1
            break

    for i in range(max_size[1]-1, -1, -1):
        if mask[0,i,0] == 1:
            col = i + 1
            break

    if row is None or col is None:
        ipdb.set_trace()
    return row, col