将 COCOeval 总结写入 tensorboard答案

【问题标题】：Write COCOeval summary to tensorboard将 COCOeval 总结写入 tensorboard
【发布时间】：2022-10-12 21:18:53
【问题描述】：

我正在使用 pycocotools 评估我的 R-CNN 模型

coco_eval = pycocotools.cocoeval.COCOeval(coco_gt)

我执行所有必要的计算，然后调用

coco_eval.accumulate()
coco_eval.summarize()

这或多或少像这样打印一个表格

 Average Precision  (AP) @[ IoU=0.50:0.95 | area=   all | maxDets=100 ] = 0.000
 Average Precision  (AP) @[ IoU=0.50      | area=   all | maxDets=100 ] = 0.000
 Average Precision  (AP) @[ IoU=0.75      | area=   all | maxDets=100 ] = 0.000
 Average Precision  (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.000
 Average Precision  (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.000
 Average Precision  (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.000
 Average Recall     (AR) @[ IoU=0.50:0.95 | area=   all | maxDets=  1 ] = 0.000
 Average Recall     (AR) @[ IoU=0.50:0.95 | area=   all | maxDets= 10 ] = 0.000
 Average Recall     (AR) @[ IoU=0.50:0.95 | area=   all | maxDets=100 ] = 0.001
 Average Recall     (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.001
 Average Recall     (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.001
 Average Recall     (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.005

有没有办法把这个写到SummaryWriter。

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter()
for category, mAP in coco_eval.summary():
    writer.add_scalar(category, mAP)

或多或少是这样的？我只能找到包含 mAP 值的coco_eval.stats，但是它们对应的类别的名称（如Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ]）在哪里？

【问题讨论】：

标签： python pytorch tensorboard coco pycocotools

【解决方案1】：

你找到解决方案了吗？我会感兴趣

【讨论】：

bjarne321，请不要添加我也是作为答案。它实际上并没有提供问题的答案。如果你有一个不同但相关的问题，那么ask它（如果它有助于提供上下文，请参考这个问题）。如果您对这个特定问题感兴趣，您可以upvote 它，留下comment，或者在您有足够的reputation 时启动bounty。