Some problems about running train and eval scripts

[ohjarwa]

HI here! Thank you for your fantastic work! However when i running the train.py script python train.py --dataset 7Scenes --scene chess --model AtLoc --gpus 0 ------------ Options ------------- batchsize: 64 beta: -3.0 color_jitter: 0.7 cropsize: 256 data_dir: ./data dataset: 7Scenes epochs: 100 exp_name: name gamma: None gpus: [0] logdir: ./logs lr: 5e-05 lstm: False model: AtLoc models_dir: models nThreads: 8 print_freq: 20 real: False results_dir: figures runs_dir: runs save_freq: 5 scene: chess seed: 7 skip: 10 steps: 3 test_dropout: 0.0 train_dropout: 0.5 val: False val_freq: 5 variable_skip: False weight_decay: 0.0005 weights: epoch_005.pth.tar -------------- End --------------- Logging to ./logs/7Scenes_chess_AtLoc_False/runs/log.txt learn_beta Using ColorJitter data augmentation Traceback (most recent call last): File "train.py", line 74, in train_set = SevenScenes(train=True, **kwargs) File "/home/ubuntu/YaoXinSheng/AtLoc-master/data/dataloaders.py", line 42, in init p_filenames = [n for n in os.listdir(osp.join(seq_dir, '.')) if n.find('pose') >= 0] FileNotFoundError: [Errno 2] No such file or directory: './data/7Scenes/chess/seq-01/.'

and when i run the eval.py ,using your trained models i get error too： python eval.py --dataset 7Scenes --scene stairs --model AtLoc --gpus 0 --weights ./logs/7Scenes_stairs_AtLoc_False/models/epoch_500.pth.tar ------------ Options ------------- batchsize: 64 beta: -3.0 color_jitter: 0.7 cropsize: 256 data_dir: ./data dataset: 7Scenes epochs: 100 exp_name: name gamma: None gpus: [0] logdir: ./logs lr: 5e-05 lstm: False model: AtLoc models_dir: models nThreads: 8 print_freq: 20 real: False results_dir: figures runs_dir: runs save_freq: 5 scene: stairs seed: 7 skip: 10 steps: 3 test_dropout: 0.0 train_dropout: 0.5 val: False val_freq: 5 variable_skip: False weight_decay: 0.0005 weights: ./logs/7Scenes_stairs_AtLoc_False/models/epoch_500.pth.tar -------------- End --------------- Traceback (most recent call last): File "eval.py", line 62, in data_set = SevenScenes(**kwargs) File "/home/ubuntu/YaoXinSheng/AtLoc-master/data/dataloaders.py", line 82, in init align_s=vo_stats[seq]['s']) File "/home/ubuntu/YaoXinSheng/AtLoc-master/tools/utils.py", line 144, in process_poses poses_out[:, 0:3] = poses_in[:, [3, 7, 11]] IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed I would appreciate if you give me your help!