search

MCC-WH / CSA

Official implementation of NeurIPS 2021 paper "Contextual Similarity Aggregation with Self-attention for Visual Re-ranking"
MIT License
25 stars 3 forks source link

train error #8

Closed WangYafei0904 closed 2 years ago

WangYafei0904 commented 2 years ago

File "G:/jiansuo code/CSA-main/train.py", line 174, in main(args=args) File "G:/jiansuo code/CSA-main/train.py", line 34, in main train_dataset = RerankDataset_TopKSIM(names=dataset_names, mode='train', topk=args.topk, sim_len=args.sim_len) File "G:\jiansuo code\CSA-main\Dataset\ImageFromList.py", line 73, in init features= torch.tensor(load_pickle(path)[mode]).float() File "G:\jiansuo code\CSA-main\utils\helpfunc.py", line 269, in load_pickle with open(path, 'rb') as file: FileNotFoundError: [Errno 2] No such file or directory: 'G:\jiansuo code\CSA-main\data\train\r.pkl'

MCC-WH commented 2 years ago

you should prepare the image features first.