| import numpy as np | |
| import cv2 | |
| def draw_proj_image(image,proj_mat,points): | |
| points_homo=np.concatenate([points,np.ones((points.shape[0],1))],axis=1) | |
| pts_inimg=np.dot(points_homo,proj_mat.T) | |
| image=cv2.resize(image,dsize=(224,224),interpolation=cv2.INTER_LINEAR) | |
| x=pts_inimg[:,0]/pts_inimg[:,2] | |
| y=pts_inimg[:,1]/pts_inimg[:,2] | |
| x=np.clip(x,a_min=0,a_max=223).astype(np.int32) | |
| y=np.clip(y,a_min=0,a_max=223).astype(np.int32) | |
| image[y,x]=np.array([[0,255,0]]) | |
| return image |