cvat-ai--cvat
11 行
267 B
Python
11 行
267 B
Python
# Copyright (C) CVAT.ai Corporation
|
|
#
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
|
|
def to_cvat_mask(box: list, mask):
|
|
xtl, ytl, xbr, ybr = box
|
|
flattened = mask[ytl : ybr + 1, xtl : xbr + 1].flat[:].tolist()
|
|
flattened.extend([xtl, ytl, xbr, ybr])
|
|
return flattened
|