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_resize_with_bbox_op_params_outside_of_interpolation_dict():
"""
Test passing in a invalid key for interpolation
"""
logger.info(
"test_resize_with_bbox_op_params_outside_of_interpolation_dict")
size = (500, 500)
more_para = None
with pytest.raises(KeyError, match="None"):
c_vision.ResizeWithBBox(size, more_para)
def test_resize_with_bbox_op_invalid_c():
"""
Test ResizeWithBBox Op on invalid constructor parameters, expected to raise ValueError
"""
logger.info("test_resize_with_bbox_op_invalid_c")
try:
# invalid interpolation value
c_vision.ResizeWithBBox(400, interpolation="invalid")
except TypeError as err:
logger.info("Got an exception in DE: {}".format(str(err)))
assert "interpolation" in str(err)
def test_resize_with_bbox_op_edge_c(plot_vis=False):
"""
Prints images and bboxes side by side with and without ResizeWithBBox Op applied,
applied on dynamically generated edge case, expected to pass. edge case is when bounding
box has dimensions as the image itself.
"""
logger.info("test_resize_with_bbox_op_edge_c")
dataVoc1 = ds.VOCDataset(DATA_DIR,
task="Detection",
mode="train",
decode=True,
shuffle=False)
dataVoc2 = ds.VOCDataset(DATA_DIR,
task="Detection",
mode="train",
decode=True,
shuffle=False)
test_op = c_vision.ResizeWithBBox(500)
# maps to convert data into valid edge case data
dataVoc1 = dataVoc1.map(
input_columns=["image", "annotation"],
output_columns=["image", "annotation"],
columns_order=["image", "annotation"],
operations=[
lambda img, bboxes:
(img, np.array([[0, 0, img.shape[1], img.shape[0]]]).astype(bboxes.
dtype))
])
dataVoc2 = dataVoc2.map(
input_columns=["image", "annotation"],
output_columns=["image", "annotation"],
columns_order=["image", "annotation"],
operations=[
lambda img, bboxes:
(img, np.array([[0, 0, img.shape[1], img.shape[0]]]).astype(
bboxes.dtype)), test_op
])
unaugSamp, augSamp = [], []
for unAug, Aug in zip(dataVoc1.create_dict_iterator(),
dataVoc2.create_dict_iterator()):
unaugSamp.append(unAug)
augSamp.append(Aug)
if plot_vis:
visualize_with_bounding_boxes(unaugSamp, augSamp)
def test_resize_with_bbox_op_coco_c(plot_vis=False):
"""
Prints images and bboxes side by side with and without ResizeWithBBox Op applied,
tests with MD5 check, expected to pass
Testing with COCO dataset
"""
logger.info("test_resize_with_bbox_op_coco_c")
# Load dataset
dataCOCO1 = ds.CocoDataset(DATA_DIR_2[0],
annotation_file=DATA_DIR_2[1],
task="Detection",
decode=True,
shuffle=False)
dataCOCO2 = ds.CocoDataset(DATA_DIR_2[0],
annotation_file=DATA_DIR_2[1],
task="Detection",
decode=True,
shuffle=False)
test_op = c_vision.ResizeWithBBox(200)
# map to apply ops
dataCOCO2 = dataCOCO2.map(input_columns=["image", "bbox"],
output_columns=["image", "bbox"],
columns_order=["image", "bbox"],
operations=[test_op])
filename = "resize_with_bbox_op_01_c_coco_result.npz"
save_and_check_md5(dataCOCO2, filename, generate_golden=GENERATE_GOLDEN)
unaugSamp, augSamp = [], []
for unAug, Aug in zip(dataCOCO1.create_dict_iterator(),
dataCOCO2.create_dict_iterator()):
unaugSamp.append(unAug)
augSamp.append(Aug)
if plot_vis:
visualize_with_bounding_boxes(unaugSamp, augSamp, annot_name="bbox")
def test_resize_with_bbox_op_voc_c(plot_vis=False):
"""
Prints images and bboxes side by side with and without ResizeWithBBox Op applied
testing with VOC dataset
"""
logger.info("test_resize_with_bbox_op_voc_c")
# Load dataset
dataVoc1 = ds.VOCDataset(DATA_DIR,
task="Detection",
mode="train",
decode=True,
shuffle=False)
dataVoc2 = ds.VOCDataset(DATA_DIR,
task="Detection",
mode="train",
decode=True,
shuffle=False)
test_op = c_vision.ResizeWithBBox(100)
# map to apply ops
dataVoc2 = dataVoc2.map(input_columns=["image", "annotation"],
output_columns=["image", "annotation"],
columns_order=["image", "annotation"],
operations=[test_op])
filename = "resize_with_bbox_op_01_c_voc_result.npz"
save_and_check_md5(dataVoc2, filename, generate_golden=GENERATE_GOLDEN)
unaugSamp, augSamp = [], []
for unAug, Aug in zip(dataVoc1.create_dict_iterator(),
dataVoc2.create_dict_iterator()):
unaugSamp.append(unAug)
augSamp.append(Aug)
if plot_vis:
visualize_with_bounding_boxes(unaugSamp, augSamp)