Skip to content

Some problems about running train and eval scripts #15

@ohjarwa

Description

@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!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions