google-research / composed_image_retrieval

Apache License 2.0
175 stars 18 forks source link

KeyError: 'filepath' - run the sample train failed & Sample evaluation result log is empty #10

Closed KeepAndWin closed 1 year ago

KeepAndWin commented 1 year ago

Hi! I tried running the train sample, but I got the errors as follows:

Traceback (most recent call last):
  File "/home/anaconda3/lib/python3.9/site-packages/torch/multiprocessing/spawn.py", line 69, in _wrap
    fn(i, *args)
  File "/home/CIR_Lab/composed_image_retrieval/src/main.py", line 126, in main_worker
    data = get_data(args, (preprocess_train, preprocess_val))
  File "/home/CIR_Lab/composed_image_retrieval/src/data.py", line 577, in get_data
    data["train"] = get_dataset_fn(args.train_data, args.dataset_type)(
  File "/home/CIR_Lab/composed_image_retrieval/src/data.py", line 473, in get_csv_dataset
    dataset = CsvDataset(
  File "/home/CIR_Lab/composed_image_retrieval/src/data.py", line 377, in __init__
    self.images = df[img_key].tolist()
  File "/home/anaconda3/lib/python3.9/site-packages/pandas/core/frame.py", line 3807, in __getitem__
    indexer = self.columns.get_loc(key)
  File "/home/anaconda3/lib/python3.9/site-packages/pandas/core/indexes/base.py", line 3804, in get_loc
    raise KeyError(key) from err
KeyError: 'filepath'

I have some questions:

KeepAndWin commented 1 year ago

@kihyuks @ksaito-ut