fan23j / yolov7-pose-whole-body

Yolov7-pose with variable keypoint support. Trained models with COCO Wholebody.
Other
10 stars 1 forks source link

Annotation Mapping Bug #7

Open fmanstein opened 11 months ago

fmanstein commented 11 months ago

I am currently trying to train this model on custom data, however when running a single epoch of training, the training_batch.jpg files indicate an error when showcasing the keypoints. They seem not to be in the bounding boxes. I suspect that the padding of not square images to be square does not work properly when changing the keypoints accordingly. Also, x and y values of keypoints appear to be swapped, as they appear to be mirrored across the x,y diagonal.

Example: for a blank 640*400 image with a single detection, with three keypoints, each located at one corner, this should be in the corresponding label file: 0 0.3 0.7 0.1 0.1 0.25 0.65 2 0.25 0.75 2 0.35 0.65 2

The resulting train_batch.jpg should be this: 0 png

However: when actually training on this single image, this image is the output: train_batch0

Is this an error in the repository code for dealing with keypoints or am I mistaken and did understand the label incorrectly?