【DenseFusion代码详解】测试过程eval_ycb.py

DenseFusion系列代码全讲解目录：【DenseFusion系列目录】代码全讲解+可视化+计算评估指标_Panpanpan！的博客-CSDN博客

这些内容均为个人学习记录，欢迎大家提出错误一起讨论一起学习！

---

该部分是对YCB数据集训练结束之后的模型进行评估，代码位置在tools/eval_ycb.py

ycb数据集上的评估指标为AUC和<2cm，但源代码里面并没有计算这两个指标的代码，这个代码里仅仅对估计出的姿态进行保存。

parser = argparse.ArgumentParser()
parser.add_argument('--dataset_root', type=str, default = '', help='dataset root dir')
parser.add_argument('--model', type=str, default = '',  help='resume PoseNet model')
parser.add_argument('--refine_model', type=str, default = '',  help='resume PoseRefineNet model')
opt = parser.parse_args()

首先是在.sh文件中可以设置的变量，有数据