merge_isomaps.py 文件源码

def show_isomap(window, isomap):
    #isomap_copy = isomap.copy()
    background = np.zeros([ISOMAP_SIZE, ISOMAP_SIZE, 4], dtype='uint8')
    background[:,:,3]=10
    mask = np.array([[int(x/8) %2==int(y/8) %2 for x in range(isomap.shape[0])] for y in range(isomap.shape[1])])
    #mask = np.array([[int(x/8) %2==0 for x in range(isomap.shape[0])] for y in range(isomap.shape[1])])
    background[mask,:3]=[200,200,200]
    mask = np.invert(mask)
    background[mask,:3]=[150,150,150]

    cv2.imshow(window, merge([background,isomap]))