如何在TensorBoard中查看更多TensorFlow目标检测的评估图像?
解决TensorBoard中评估图像数量受限的问题
我之前也碰到过一模一样的情况——训练和评估跑起来都正常,但TensorBoard里就只显示10张评估图,改max_evals完全没效果,因为这个参数根本不是干这个的!它控制的是评估任务重复执行的次数,和显示多少张图像没关系。
要增加TensorBoard里能查看的评估图像数量,你需要调整以下两个关键配置:
修改
eval_config中的num_visualizations参数
这个参数就是专门控制TensorBoard中导出并显示的评估图像数量的,默认值是10。你可以把它设成你想要的数字,比如50或者100。确保
num_examples足够大
你已经设置了num_examples: 1000,这个没问题,但要保证它的值大于等于你设置的num_visualizations,否则系统最多只能显示num_examples数量的图像。
修改后的配置文件片段应该是这样的:
eval_config: { num_examples: 1000 max_evals: 50 num_visualizations: 50 # 这里改成你想要的图像数量 } eval_input_reader: { tf_record_input_reader { input_path: "xxx/eval.record" } label_map_path: "xxx/label_map.pbtxt" shuffle: false num_readers: 1 }
另外还要注意两点:
- 修改配置后,需要重启评估任务,新的配置才会生效
- 如果你用的是比较旧的TensorFlow Object Detection API版本,这个参数可能叫
max_visualizations,如果num_visualizations没效果,可以试试这个参数名
调整完重新运行评估,TensorBoard里就能看到更多的评估图像了。
内容的提问来源于stack exchange,提问作者jood
相关产品推荐
相关产品推荐

