def test_filter_parts(self): fig, ax = plt.subplots(1, 2, figsize=(10, 5)) kps = stickman.VUNetStickman.get_example_valid_keypoints_deepfashion() box = stickman.get_bounding_boxes(kps, (128, 128), 32) joint_img = stickman.VUNetStickman.make_joint_img( (128, 128), stickman.VUNET_JOINT_ORDER_DEEPFASHION, stickman.VUNetStickman.get_example_valid_keypoints_deepfashion() * 128, ) box_image = plotting.overlay_boxes_without_labels(joint_img, box) ax[0].imshow(box_image) kps = stickman.VUNetStickman.get_example_valid_keypoints_deepfashion() jo = ["lankle", "rankle"] new_kps = stickman.filter_keypoints( kps, stickman.VUNET_JOINT_ORDER_DEEPFASHION, jo) box = stickman.get_bounding_boxes(new_kps, (128, 128), 32) # joint_img = stickman.VUNetStickman.make_joint_img((128, 128), jo, new_kps * 128) box_image = plotting.overlay_boxes_without_labels(joint_img, box, colors=[ [0, 0, 255], ] * len(new_kps)) ax[1].imshow(box_image) return plt.gcf()
def test_overlay_boxes_without_labels(self): fig, ax = plt.subplots(1, 1) image = skimage.data.astronaut() bboxes = [np.array([0, 0, 50, 50])] overlay = smplt.overlay_boxes_without_labels(image, bboxes) ax.imshow(overlay) return fig
def test_invalid_stickman(self): kps = stickman.VUNetStickman.get_example_invalid_keypoints() box = stickman.get_bounding_boxes(kps, (128, 128), 32) joint_img = stickman.VUNetStickman.make_joint_img( (128, 128), stickman.VUNET_JOINT_ORDER, stickman.VUNetStickman.get_example_invalid_keypoints() * 128, ) box_image = plotting.overlay_boxes_without_labels(joint_img, box) plt.imshow(box_image) return plt.gcf()
def test_overlay_boxes_without_labels(self): fig, ax = plt.subplots(1, 1) import skimage image = skimage.data.astronaut() bboxes = [np.array([0, 0, 50, 50])] from supermariopy import plotting overlay = plotting.overlay_boxes_without_labels(image, bboxes) ax.imshow(overlay) return fig