| #!/bin/bash
|
|
|
| set -x
|
| set -e
|
|
|
| export PYTHONUNBUFFERED="True"
|
| export CUDA_VISIBLE_DEVICES=0
|
|
|
| if [ ! -d YCB_Video_toolbox ];then
|
| echo 'Downloading the YCB_Video_toolbox...'
|
| git clone https://github.com/yuxng/YCB_Video_toolbox.git
|
| cd YCB_Video_toolbox
|
| unzip results_PoseCNN_RSS2018.zip
|
| cd ..
|
| cp replace_ycb_toolbox/*.m YCB_Video_toolbox/
|
| fi
|
|
|
| python3 ./tools/eval_ycb.py --dataset_root ./datasets/ycb/YCB_Video_Dataset\
|
| --model trained_checkpoints/ycb/pose_model_26_0.012863246640872631.pth\
|
| --refine_model trained_checkpoints/ycb/pose_refine_model_69_0.009449292959118935.pth |
