def test_convert_keypoints_from_albumentations(): keypoints = [[0.2, 0.3, 0.6, 0.8], [0.3, 0.4, 0.7, 0.9, 99]] image = np.ones((100, 100, 3)) converted_keypointes = convert_keypoints_from_albumentations(keypoints, rows=image.shape[0], cols=image.shape[1], target_format='xyas') converted_keypoint_1 = convert_keypoint_from_albumentations(keypoints[0], rows=image.shape[0], cols=image.shape[1], target_format='xyas') converted_keypoint_2 = convert_keypoint_from_albumentations(keypoints[1], rows=image.shape[0], cols=image.shape[1], target_format='xyas') assert converted_keypointes == [converted_keypoint_1, converted_keypoint_2]
def test_convert_keypoint_to_albumentations_and_back(kp, keypoint_format): image = np.ones((100, 100, 3)) converted_kp = convert_keypoint_to_albumentations(kp, rows=image.shape[0], cols=image.shape[1], source_format=keypoint_format) converted_back_kp = convert_keypoint_from_albumentations(converted_kp, rows=image.shape[0], cols=image.shape[1], target_format=keypoint_format) assert converted_back_kp == kp
def test_convert_keypoint_from_albumentations(kp, target_format, expected): image = np.ones((100, 100, 3)) converted_keypoint = convert_keypoint_from_albumentations( kp, rows=image.shape[0], cols=image.shape[1], target_format=target_format) assert converted_keypoint == expected