def test_resize_with_bbox_op_bad_c():
"""
Tests ResizeWithBBox Op with invalid bounding boxes, expected to catch multiple errors
"""
logger.info("test_resize_with_bbox_op_bad_c")
test_op = c_vision.ResizeWithBBox((200, 300))
data_voc2 = ds.VOCDataset(DATA_DIR,
task="Detection",
mode="train",
decode=True,
shuffle=False)
check_bad_bbox(data_voc2, test_op, InvalidBBoxType.WidthOverflow,
"bounding boxes is out of bounds of the image")
data_voc2 = ds.VOCDataset(DATA_DIR,
task="Detection",
mode="train",
decode=True,
shuffle=False)
check_bad_bbox(data_voc2, test_op, InvalidBBoxType.HeightOverflow,
"bounding boxes is out of bounds of the image")
data_voc2 = ds.VOCDataset(DATA_DIR,
task="Detection",
mode="train",
decode=True,
shuffle=False)
check_bad_bbox(data_voc2, test_op, InvalidBBoxType.NegativeXY, "min_x")
data_voc2 = ds.VOCDataset(DATA_DIR,
task="Detection",
mode="train",
decode=True,
shuffle=False)
check_bad_bbox(data_voc2, test_op, InvalidBBoxType.WrongShape,
"4 features")
def test_bounding_box_augment_invalid_bounds_c():
"""
Test BoundingBoxAugment op with invalid bboxes.
"""
logger.info("test_bounding_box_augment_invalid_bounds_c")
test_op = c_vision.BoundingBoxAugment(c_vision.RandomHorizontalFlip(1), 1)
dataVoc2 = ds.VOCDataset(DATA_DIR,
task="Detection",
mode="train",
decode=True,
shuffle=False)
check_bad_bbox(dataVoc2, test_op, InvalidBBoxType.WidthOverflow,
"bounding boxes is out of bounds of the image")
dataVoc2 = ds.VOCDataset(DATA_DIR,
task="Detection",
mode="train",
decode=True,
shuffle=False)
check_bad_bbox(dataVoc2, test_op, InvalidBBoxType.HeightOverflow,
"bounding boxes is out of bounds of the image")
dataVoc2 = ds.VOCDataset(DATA_DIR,
task="Detection",
mode="train",
decode=True,
shuffle=False)
check_bad_bbox(dataVoc2, test_op, InvalidBBoxType.NegativeXY, "min_x")
dataVoc2 = ds.VOCDataset(DATA_DIR,
task="Detection",
mode="train",
decode=True,
shuffle=False)
check_bad_bbox(dataVoc2, test_op, InvalidBBoxType.WrongShape, "4 features")
def test_random_vertical_flip_with_bbox_op_bad_c():
"""
Tests RandomVerticalFlipWithBBox Op with invalid bounding boxes, expected to catch multiple errors
"""
logger.info("test_random_vertical_flip_with_bbox_op_bad_c")
test_op = c_vision.RandomVerticalFlipWithBBox(1)
data_voc2 = ds.VOCDataset(DATA_DIR_VOC,
task="Detection",
usage="train",
shuffle=False,
decode=True)
check_bad_bbox(data_voc2, test_op, InvalidBBoxType.WidthOverflow,
"bounding boxes is out of bounds of the image")
data_voc2 = ds.VOCDataset(DATA_DIR_VOC,
task="Detection",
usage="train",
shuffle=False,
decode=True)
check_bad_bbox(data_voc2, test_op, InvalidBBoxType.HeightOverflow,
"bounding boxes is out of bounds of the image")
data_voc2 = ds.VOCDataset(DATA_DIR_VOC,
task="Detection",
usage="train",
shuffle=False,
decode=True)
check_bad_bbox(data_voc2, test_op, InvalidBBoxType.NegativeXY,
"negative value")
data_voc2 = ds.VOCDataset(DATA_DIR_VOC,
task="Detection",
usage="train",
shuffle=False,
decode=True)
check_bad_bbox(data_voc2, test_op, InvalidBBoxType.WrongShape,
"4 features")
def test_random_horizontal_flip_with_bbox_invalid_bounds_c():
"""
Test RandomHorizontalFlipWithBBox op with invalid bounding boxes
"""
logger.info("test_random_horizontal_bbox_invalid_bounds_c")
test_op = c_vision.RandomHorizontalFlipWithBBox(1)
dataVoc2 = ds.VOCDataset(DATA_DIR,
task="Detection",
usage="train",
shuffle=False,
decode=True)
check_bad_bbox(dataVoc2, test_op, InvalidBBoxType.WidthOverflow,
"bounding boxes is out of bounds of the image")
dataVoc2 = ds.VOCDataset(DATA_DIR,
task="Detection",
usage="train",
shuffle=False,
decode=True)
check_bad_bbox(dataVoc2, test_op, InvalidBBoxType.HeightOverflow,
"bounding boxes is out of bounds of the image")
dataVoc2 = ds.VOCDataset(DATA_DIR,
task="Detection",
usage="train",
shuffle=False,
decode=True)
check_bad_bbox(dataVoc2, test_op, InvalidBBoxType.NegativeXY,
"negative value")
dataVoc2 = ds.VOCDataset(DATA_DIR,
task="Detection",
usage="train",
shuffle=False,
decode=True)
check_bad_bbox(dataVoc2, test_op, InvalidBBoxType.WrongShape, "4 features")
def test_random_resized_crop_with_bbox_op_bad_c():
"""
Test RandomCropWithBBox op with invalid bounding boxes, expected to catch multiple errors.
"""
logger.info("test_random_resized_crop_with_bbox_op_bad_c")
test_op = c_vision.RandomResizedCropWithBBox((256, 512), (0.5, 0.5),
(0.5, 0.5))
data_voc2 = ds.VOCDataset(DATA_DIR_VOC,
task="Detection",
usage="train",
shuffle=False,
decode=True)
check_bad_bbox(data_voc2, test_op, InvalidBBoxType.WidthOverflow,
"bounding boxes is out of bounds of the image")
data_voc2 = ds.VOCDataset(DATA_DIR_VOC,
task="Detection",
usage="train",
shuffle=False,
decode=True)
check_bad_bbox(data_voc2, test_op, InvalidBBoxType.HeightOverflow,
"bounding boxes is out of bounds of the image")
data_voc2 = ds.VOCDataset(DATA_DIR_VOC,
task="Detection",
usage="train",
shuffle=False,
decode=True)
check_bad_bbox(data_voc2, test_op, InvalidBBoxType.NegativeXY, "min_x")
data_voc2 = ds.VOCDataset(DATA_DIR_VOC,
task="Detection",
usage="train",
shuffle=False,
decode=True)
check_bad_bbox(data_voc2, test_op, InvalidBBoxType.WrongShape,
"4 features")
