def TestDatasetCreator(): subject_name = "davide1" dc = DatasetCreator(config.folder_path + "/data/Hand/" + subject_name + ".bag") start_frame, end_frame = dc.FrameSelector() dc.CreateBebopDataset( 0, config.folder_path + "/data/Hand/" + subject_name + "Hand.pickle", start_frame, end_frame) subject_name = "davide2" dc2 = DatasetCreator(config.folder_path + "/data/Hand/" + subject_name + ".bag") start_frame, end_frame = dc2.FrameSelector() dc2.CreateBebopDataset( 0, config.folder_path + "/data/Hand/" + subject_name + "Hand.pickle", start_frame, end_frame) folderPath = config.folder_path + "/data/Hand/" fileList = ["davide1Hand.pickle", "davide2Hand.pickle"] DatasetCreator.JoinPickleFiles( fileList, config.folder_path + "/data/Hand/DavideHand.pickle", folderPath)
def JoinPickles(fileList, picklename): picklefolder = config.folder_path + "/../Pickles/16_4_2020/" #fileList = {"Clip1.pickle", "Clip2.pickle", "Clip3.pickle", "Clip4.pickle", "Clip5.pickle", "Clip6.pickle"} DatasetCreator.JoinPickleFiles(fileList, picklefolder + picklename, picklefolder)