search

wenbowen123 / iros20-6d-pose-tracking

[IROS 2020] se(3)-TrackNet: Data-driven 6D Pose Tracking by Calibrating Image Residuals in Synthetic Domains
Other
385 stars 66 forks source link

about predict.sh #10

Closed baojun701 closed 3 years ago

baojun701 commented 3 years ago

I update the path in the predict.sh, but it doesn't work.

ckpt_dir: None dataset_info_path None/../dataset_info.yml Traceback (most recent call last): File "predict.py", line 646, in with open(dataset_info_path,'r') as ff: FileNotFoundError: [Errno 2] No such file or directory: 'None/../dataset_info.yml'

here is my predict.sh: CUDA_VISIBLE_DEVICES=1,2,3,4 \ python /home/zkyd/iros20-6d-pose-tracking-master/predict.py \ --train_data_path /home/zkyd/iros20-6d-pose-tracking-master/data/YCB_Video_Dataset/data_synthetic/banana/train_data_blender_DR \ --ckpt_dir /home/zkyd/iros20-6d-pose-tracking-master/data/YCB_weights/banana/model_best_val.pth.tar \ --mean_std_path /home/zkyd/iros20-6d-pose-tracking-master/data/YCB_weights/banana \ --class_id 3 \ why?

wenbowen123 commented 3 years ago

Did you pass "None" to ckpt_dir? You previous path configuration is correct. You only need to download the correct data. See https://github.com/wenbowen123/iros20-6d-pose-tracking/issues/9

baojun701 commented 3 years ago

I want to modify the path in the predict.sh to test other object, but it doesn't work. I modify the path in the predict.py(default), it's OK, I want to know why? the predict.sh is useless? I account "None" when I run.

wenbowen123 commented 3 years ago

It's likely that you path is not set correctly here, you need to change to your path to predict.py.