the problem about the dataset

Tom-Teamo commented 2 years ago

In "transformer-csi.py", there is parser.add_argument('--dataset', type=str, default='53001_npy', help='dataset'). I am confused by "53001_npy", and I wonder whether the dataset(https://drive.google.com/file/d/19uH0_z1MBLtmMLh8L4BlNA0w-XAFKipM/view) is in 53001_npy folder? By the way, I think that "load_public_data.py" is used to torch.save(data, "Data.pt"), right? If I run this script, does it mean that I do not need to run below IF segment of "transformer-csi.py", because I already hava "Data.pt" after running.

def load_data(root):
    if root.find("npy") != -1:
        root = root + '\\'
        file_list = os.listdir(root)
        label = []
        data = []
        aclist = ['empty', 'jump', 'pick', 'run', 'sit', 'walk', 'wave']
        for file in file_list:
            file_name = root + file
            csi = np.load(file_name)
            csi = torch.from_numpy(csi).float().unsqueeze(0)  # 1*2000*3*30
            csi.requires_grad = False
            csi = csi.view(1, 2000, 90)
            data.append(csi)
            for j in range(len(aclist)):
                if file.find(aclist[j]) != -1:
                    label.append(j)
                    break
        data = torch.cat(data, dim=0)
        label = torch.tensor(label)
        data = Data.TensorDataset(data, label)
        args.category = len(aclist)
    else:
        data = torch.load("Data.pt")
        aclist = ['bed', 'fall', 'pickup', 'run', 'sitdown', 'standup', 'walk']
    return data, aclist