Shr3ezy's picture
Uploading all files
29858c0 verified
raw
history blame contribute delete
617 Bytes
#!/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