TexasInstruments / edgeai-yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite. Forked from https://ultralytics.com/yolov5
https://github.com/TexasInstruments/edgeai
GNU General Public License v3.0
641 stars 121 forks source link

On the question of keypoint ap calculation #117

Open Hongru0306 opened 1 year ago

Hongru0306 commented 1 year ago

🚀 Feature

Motivation

While reading the ap calculation section of your pose branch source code, I noticed that you seem to be using bounding boxes for the ap calculation, rather than keypoints.

Pitch

Pose branching, key point ap calculation

Alternatives

In looking at the code for your paper, is it biased that the test script inside the yolo-pose branch finds ap calculations, instead of using ap calculations for keypoints, it uses ap calculations for the person prediction box? b43435871fcdf44bb1c5fa9914dd48d

Additional context

I hope the author can reply in his busy schedule, here for the oks ap calculation should be how to set up ah!

atharvavaidya14 commented 1 year ago

Hi, I also have this same issue and I want to include the precision, recall and mAPs for the pose estimation part (just like they do in yolov8-pose). Have you found a solution to this?

Hongru0306 commented 1 year ago

Hi, I also have this same issue and I want to include the precision, recall and mAPs for the pose estimation part (just like they do in yolov8-pose). Have you found a solution to this?

I'm trying to write it myself, but the accuracy results never agree with what's on the paper, so it's too sad that there should be something wrong with my writing.

atharvavaidya14 commented 1 year ago

Can you give a short guide/code on how to integrate it? Will be really helpful for me. Especially the part in test.py where we calculate the total true positives for keypoints (using OKS?).

Hongru0306 commented 1 year ago

Can you give a short guide/code on how to integrate it? Will be really helpful for me. Especially the part in test.py where we calculate the total true positives for keypoints (using OKS?).

This is a bit long, a few words can't be very clear, the function and calculation method for calculating ap in test.py is exactly the same as yolo, you can check how yolo calculates it first. You can check how yolo calculates it first. If you have any questions after reading it, we can discuss it again.

MR-STUZHANG commented 7 months ago

您好,我也发现了这个问题,您解决了吗。

luoriyanguilai commented 4 months ago

你好,请问你解决了吗?

MR-STUZHANG commented 4 months ago

解决了 ---- 回复的原邮件 ---- 发件人 @.> 日期 2024年04月21日 19:12 收件人 @.> 抄送至 @.>@.> 主题 Re: [TexasInstruments/edgeai-yolov5] On the question of keypoint ap calculation (Issue #117) 你好,请问你解决了吗? — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you commented.Message ID: @.***>

Hongru0306 commented 4 months ago

这个repo是用cocotools来算的

Song-Jie-ing commented 2 weeks ago

解决了 ---- 回复的原邮件 ---- 发件人 @.> 日期 2024年04月21日 19:12 收件人 @.> 抄送至 @.>@.> 主题 Re: [TexasInstruments/edgeai-yolov5] On the question of keypoint ap calculation (Issue #117) 你好,请问你解决了吗? — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you commented.Message ID: @.***>

你好!请问针对yolopose中它没有关键点的map计算这个问题应该怎么解决能?这个问题真的困扰我好a久。

Song-Jie-ing commented 2 weeks ago

这个repo是用cocotools来算的

楼主,这个yolopose中它没有关键点的map计算这个问题解决了吗?

Hongru0306 commented 2 weeks ago

这个repo是用cocotools来算的

楼主,这个yolopose中它没有关键点的map计算这个问题解决了吗?

可以看我上一条回复,仔细看一下它test的源码,它是用cocotools的组件计算的。

Song-Jie-ing commented 2 weeks ago

这个repo是用cocotools来算的

楼主,这个yolopose中它没有关键点的map计算这个问题解决了吗?

可以看我上一条回复,仔细看一下它test的源码,它是用cocotools的组件计算的。

            # from pycocotools.coco import COCO
            # from pycocotools.cocoeval import COCOeval

            from xtcocotools.coco import COCO
            from xtcocotools.cocoeval import COCOeval

是不是这四句代码,把上面两句注释的代码注释回来,然后把下面两句代码注释掉

Hongru0306 commented 2 weeks ago

这个repo是用cocotools来算的

楼主,这个yolopose中它没有关键点的map计算这个问题解决了吗?

可以看我上一条回复,仔细看一下它test的源码,它是用cocotools的组件计算的。

            # from pycocotools.coco import COCO
            # from pycocotools.cocoeval import COCOeval

            from xtcocotools.coco import COCO
            from xtcocotools.cocoeval import COCOeval

是不是这四句代码,把上面两句注释的代码注释回来,然后把下面两句代码注释掉

是的,就在这个位置

Song-Jie-ing commented 2 weeks ago

这个repo是用cocotools来算的

楼主,这个yolopose中它没有关键点的map计算这个问题解决了吗?

可以看我上一条回复,仔细看一下它test的源码,它是用cocotools的组件计算的。

            # from pycocotools.coco import COCO
            # from pycocotools.cocoeval import COCOeval

            from xtcocotools.coco import COCO
            from xtcocotools.cocoeval import COCOeval

是不是这四句代码,把上面两句注释的代码注释回来,然后把下面两句代码注释掉

是的,就在这个位置

可是我在这个位置写pirnt,训练的时候并没有打印出什么东西来,好像代码就没有运行到这里。 兄弟,能不能要你个联系方式,请教下。

Hongru0306 commented 2 weeks ago

这个repo是用cocotools来算的

楼主,这个yolopose中它没有关键点的map计算这个问题解决了吗?

可以看我上一条回复,仔细看一下它test的源码,它是用cocotools的组件计算的。

            # from pycocotools.coco import COCO
            # from pycocotools.cocoeval import COCOeval

            from xtcocotools.coco import COCO
            from xtcocotools.cocoeval import COCOeval

是不是这四句代码,把上面两句注释的代码注释回来,然后把下面两句代码注释掉

是的,就在这个位置

可是我在这个位置写pirnt,训练的时候并没有打印出什么东西来,好像代码就没有运行到这里。 兄弟,能不能要你个联系方式,请教下。

训练的时候没有运行这一块,好像只有测试的时候用到了。时间有些久了我记得也不是很清了

Song-Jie-ing commented 2 weeks ago

这个repo是用cocotools来算的

楼主,这个yolopose中它没有关键点的map计算这个问题解决了吗?

可以看我上一条回复,仔细看一下它test的源码,它是用cocotools的组件计算的。

            # from pycocotools.coco import COCO
            # from pycocotools.cocoeval import COCOeval

            from xtcocotools.coco import COCO
            from xtcocotools.cocoeval import COCOeval

是不是这四句代码,把上面两句注释的代码注释回来,然后把下面两句代码注释掉

是的,就在这个位置

可是我在这个位置写pirnt,训练的时候并没有打印出什么东西来,好像代码就没有运行到这里。 兄弟,能不能要你个联系方式,请教下。

训练的时候没有运行这一块,好像只有测试的时候用到了。时间有些久了我记得也不是很清了

意思是不是yolopose的map计算的本来就是关键点的,不需要改什么东西是吗?