首页 Paddle框架 帖子详情
训练时开启验证,在第二次验证时报错KeyError2 已解决
收藏
快速回复
Paddle框架 问答模型训练 2108 4
训练时开启验证,在第二次验证时报错KeyError2 已解决
收藏
快速回复
Paddle框架 问答模型训练 2108 4

在训练时开启验证,第一次验证没问题,第二次验证报错,这是什么问题?

3# 回复于2022-03
KeyError: 2 可能是个字典,字典中没有2这个名字 建议你打印一下label_to_cat_id_map看看
0
收藏
回复
全部评论(4)
时间顺序
Z
ZXY凌漫
#2 回复于2022-03

[03/09 16:47:50] ppdet.engine INFO: Epoch: [9] [350/375] learning_rate: 0.000931 loss_xy: 7.253371 loss_wh: 3.121379 loss_obj: 17.600906 loss_cls: 0.930968 loss: 29.324682 eta: 3:36:09 batch_cost: 0.8460 data_cost: 0.5302 ips: 9.4559 images/s
[03/09 16:48:14] ppdet.utils.checkpoint INFO: Save checkpoint: output/yolov3_darknet53_270e_coco
[03/09 16:48:14] ppdet.engine INFO: Eval iter: 0
Traceback (most recent call last):
File "tools/train.py", line 177, in
main()
File "tools/train.py", line 173, in main
run(FLAGS, cfg)
File "tools/train.py", line 127, in run
trainer.train(FLAGS.eval)
File "/home/ubuntu/lxd-storage/zhangxinyun/PaddleCV/PaddleDetection/ppdet/engine/trainer.py", line 453, in train
self._eval_with_loader(self._eval_loader)
File "/home/ubuntu/lxd-storage/zhangxinyun/PaddleCV/PaddleDetection/ppdet/engine/trainer.py", line 479, in _eval_with_loader
metric.update(data, outs)
File "/home/ubuntu/lxd-storage/zhangxinyun/PaddleCV/PaddleDetection/ppdet/metrics/metrics.py", line 106, in update
outs, self.clsid2catid, bias=self.bias)
File "/home/ubuntu/lxd-storage/zhangxinyun/PaddleCV/PaddleDetection/ppdet/metrics/coco_utils.py", line 53, in get_infer_results
outs['bbox'], outs['bbox_num'], im_id, catid, bias=bias)
File "/home/ubuntu/lxd-storage/zhangxinyun/PaddleCV/PaddleDetection/ppdet/metrics/json_results.py", line 30, in get_det_res
category_id = label_to_cat_id_map[int(num_id)]
KeyError: 2

0
回复
奔向未来的样子
#3 回复于2022-03

KeyError: 2

可能是个字典,字典中没有2这个名字

建议你打印一下label_to_cat_id_map看看

0
回复
留取丹心lhqlhq
#5 回复于2023-11

遇到了同样的问题,请问大佬怎么解决的

 

0
回复
抱明月而长终v
#6 回复于2024-10

category_id = label_to_cat_id_map[int(num_id)]

就是这里突然有个num_id在map里面找不到了,这个怎么解决的啊,是数据问题吗

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户