edgeai-yolov5 icon indicating copy to clipboard operation
edgeai-yolov5 copied to clipboard

Published 20 hours ago • verified publisher icon TexasInstruments

Wrong kpts in train batch

Open Michelvl92 opened this issue 2 years ago • 3 comments

Hi, I sometimes have wrong keypoints in a training batch when training on the included coco_kpts.yaml file and default labels as downloaded from the repo, see attached img.

train_batch0

I used the latest version of the repo.

Michelvl92 avatar Sep 16 '22 08:09 Michelvl92

I encountered the same problem, this issue may help you

Gwencong avatar Sep 23 '22 03:09 Gwencong

@Gwencong I already implemented this, but it doesn't help

Michelvl92 avatar Oct 12 '22 11:10 Michelvl92

https://github.com/TexasInstruments/edgeai-yolov5/blob/ae4e0e8b399959f0b60b0f555d4aafc49ae1d859/utils/plots.py#L109

You need to modify the function of 'plot_skeleton_kpts' in 'plots.py'. Since it is hard-coded with an image size of 640, it will be resolved by changing the image size that was entered for the parameter value when training.

JoonHoonKim avatar Oct 13 '22 08:10 JoonHoonKim