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
