def figure1n2(info,path): if not os.path.exists(path): os.makedirs(path) sessions = info.groupby(level=['subject','session']) for (subject,session),sinfo in sessions: fig = plt.figure() subjectpath = os.path.join(activitymovies.datafolder,subject) sact = activitytables.read_subjects(subjectpath,days=[session]) scr = activitytables.read_subjects(subjectpath,days=[session], selector = activitytables.crossings) slipframes = activitytables.slipframes(sact,scr,sinfo,3,3) side = int(np.ceil(np.sqrt(len(slipframes)))) tile = imgproc.tile(slipframes,side,side) plt.imshow(tile[0], cmap = plt.cm.Greys_r) plt.title(str.format('{0} (session {1})',subject,session)) fname = str.format("{0}_session_{1}_slip_posture.png", subject, session) fpath = os.path.join(path,subject) if not os.path.exists(fpath): os.makedirs(fpath) fpath = os.path.join(fpath,fname) plt.axis("off") plt.savefig(fpath) plt.close(fig)
def showcluster(frames, clusters, K): for k in range(K): cluster = [ frames[i] for i in range(len(frames)) if clusters[i, 0] == k ] tiles = imgproc.tile(cluster, 8, 12) cv2.imshow('cluster{0}'.format(k), tiles[0])
def figure1n2(info, path): if not os.path.exists(path): os.makedirs(path) sessions = info.groupby(level=['subject', 'session']) for (subject, session), sinfo in sessions: fig = plt.figure() subjectpath = os.path.join(activitymovies.datafolder, subject) sact = activitytables.read_subjects(subjectpath, days=[session]) scr = activitytables.read_subjects(subjectpath, days=[session], selector=activitytables.crossings) slipframes = activitytables.slipframes(sact, scr, sinfo, 3, 3) side = int(np.ceil(np.sqrt(len(slipframes)))) tile = imgproc.tile(slipframes, side, side) plt.imshow(tile[0], cmap=plt.cm.Greys_r) plt.title(str.format('{0} (session {1})', subject, session)) fname = str.format("{0}_session_{1}_slip_posture.png", subject, session) fpath = os.path.join(path, subject) if not os.path.exists(fpath): os.makedirs(fpath) fpath = os.path.join(fpath, fname) plt.axis("off") plt.savefig(fpath) plt.close(fig)
def __tileclips__(frames,width,height): for frame in frames: yield imgproc.tile(frame,width,height)[0]
def showcluster(frames,clusters,K): for k in range(K): cluster = [frames[i] for i in range(len(frames)) if clusters[i,0] == k] tiles = imgproc.tile(cluster,8,12) cv2.imshow('cluster{0}'.format(k),tiles[0])
def __tileclips__(frames, width, height): for frame in frames: yield imgproc.tile(frame, width, height)[0]