Exemplo n.º 1
0
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)