Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

I encountered a problem exporting YOLO using the label studio annotation tool Provide posture estimation training prompts ignoring corrupt image/label: labels require 7 columns each #6754

Open
1030907690 opened this issue Dec 4, 2024 · 1 comment

Comments

@1030907690
Copy link

Describe the bug
I encountered a problem exporting YOLO using the label studio annotation tool Provide posture estimation training prompts

ignoring corrupt image/label: labels require 7 columns each

ValueError: not enough values to unpack (expected 3, got 0)

I saw the official YOLO provided example coco8-pose (example address https://github.com/ultralytics/assets/releases/download/v0.0.0/coco8-pose.zip )The label has 56 columns, so I think posture training should be related to the number of columns.

I found that the label txt generated by exporting YOLO using the label studio annotation tool always has 5 columns. How can I solve this problem?

To Reproduce
Steps to reproduce the behavior:

  1. Export YOLO from Lab Studio
  2. YAML configuration
path: D:/work/self/ultralytics/ultralytics/cfg/datasets/dinosaur_pose # dataset root dir
train: train/images # train images (relative to 'path') 4 images
val: train/images # val images (relative to 'path') 4 images
test: # test images (optional)

# Keypoints
kpt_shape: [1, 2] # number of keypoints, number of dims (2 for x,y or 3 for x,y,visible)
 
# Classes
names:
  0: hero
  1: nose

3.Training, abnormal after running

from ultralytics import YOLO
model = YOLO("yolo11n-pose.pt")  # load a pretrained model (recommended for training)
# Train the model
results = model.train(data="dinosaur-pose.yaml", epochs=100, imgsz=640)
  1. See error

image

image

Expected behavior
Do not warn or report errors.

Screenshots

image

image

Environment (please complete the following information):

  • OS: Windows 10
  • Label Studio Version 1.14.0.post0

Additional context
Only one key point, key point nose, runtime prompt

@makseq
Copy link
Member

makseq commented Dec 9, 2024

It seems you use keypoints in your project, however, LS doesn't support export to YOLO keypoint format.

However, you can use this SDK code to do it:
#6502

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants