%matplotlib qt
import numpy as np
import matplotlib.pyplot as plt
from mtflearn import ZPs
from mtflearn.utils import normalize_image
from skimage.transform import resize
filename = 'data\\monolayer_MoSe2_80K.npy'
img = np.load(filename)[0:800, 0:800]
img = resize(img, (512, 512), order=0)
img = normalize_image(img)
zps = ZPs(n_max=10, size=72)
m = zps.fit_transform(img)
m3, m6 = m.rot_maps([3, 6])
hh = m.mirror_map(m_unselect=[0,1])
plt.imshow(hh[40:-40, 40:-40])
plt.imshow(m3[40:-40, 40:-40])