def show_pts_on_img(image, pts, img_pth = '/home/workspace2/dataset/3dhand/dataset/pts_on_img.png'): if pts.shape[1] == 3: kps = [Keypoint(x, y) for x,y,_ in pts] else : kps = [Keypoint(x, y) for x,y in pts] kpsoi = KeypointsOnImage(kps, shape=image.shape) misc.imsave('/home/workspace2/dataset/3dhand/dataset/img.png', image) misc.imsave(img_pth, kpsoi.draw_on_image(image, size = 7))
#https://nbviewer.jupyter.org/github/aleju/imgaug-doc/blob/master/notebooks/B02%20-%20Augment%20Bounding%20Boxes.ipynb import imageio import imgaug as ia from imgaug.augmentables.kps import Keypoint, KeypointsOnImage %matplotlib inline #Macropus_rufogriseus_rufogriseus_Bruny="https://upload.wikimedia.org/wikipedia/commons/e/e6/Macropus_rufogriseus_rufogriseus_Bruny.jpg" Macropus_rufogriseus_rufogriseus_Bruny="D:/project/me_cfar10/imgaug_data/Macropus_rufogriseus_rufogriseus_Bruny.jpg" image = imageio.imread(Macropus_rufogriseus_rufogriseus_Bruny) image = ia.imresize_single_image(image, (389, 259)) #ia.imshow(image) #keypoint位置 kps = [ Keypoint(x=99, y=81), # left eye (from camera perspective) Keypoint(x=125, y=80), # right eye Keypoint(x=112, y=102), # nose Keypoint(x=102, y=210), # left paw Keypoint(x=127, y=207) # right paw ] #得到keypoint on image 的信息, 经过图像的增广,他们的相对位置不变, 必须图片对应 kpsoi = KeypointsOnImage(kps, shape=image.shape) ia.imshow(kpsoi.draw_on_image(image, size=7)) print(kpsoi.keypoints)#可以打印图片的keypoint 位置信息 #keypoint 可以进行平移 kpsoi_pad = kpsoi.shift(x=100) ia.imshow(kpsoi_pad.draw_on_image(image, size=7, color =(0, 255, 0)))