-
Notifications
You must be signed in to change notification settings - Fork 18
Description
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!