JoonHoonKim

> Have you solved it? This is a problem that arises because the elements being computed go into the cpu and gpu respectively. At line 531 of the 'general.py' file,...

https://github.com/TexasInstruments/edgeai-yolov5/blob/ae4e0e8b399959f0b60b0f555d4aafc49ae1d859/utils/plots.py#L504 Have you checked the 'plot_results' function? When learning key points, the value index number is different when the 'result.txt' file is loaded.

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...