WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
GNU General Public License v3.0
13.27k stars 4.19k forks source link

YOLOV7-pose algorithm. #1110

Open aoqiangma opened 1 year ago

aoqiangma commented 1 year ago

This error occurred when I ran the YOLOV7-pose algorithm.

train: Scanning 'F:\code\yolov7-pose\point\train2017' images and labels... 18 found, 0 missing, 0 empty, 18 corrupted: 100%|██████████| 18/18 [00:00<00:00, 1199.57it/s] train: New cache created: F:\code\yolov7-pose\point\train2017.cache Traceback (most recent call last): File "F:/code/yolov7-pose/train.py", line 562, in train(hyp, opt, device, tb_writer) File "F:/code/yolov7-pose/train.py", line 204, in train image_weights=opt.image_weights, quad=opt.quad, prefix=colorstr('train: '), kpt_label=kpt_label) File "F:\code\yolov7-pose\utils\datasets.py", line 74, in create_dataloader kpt_label=kpt_label) File "F:\code\yolov7-pose\utils\datasets.py", line 414, in init labels, shapes, self.segments = zip(*cache.values()) ValueError: not enough values to unpack (expected 3, got 0) train: WARNING: Ignoring corrupted image and/or label F:\code\yolov7-pose\point\images\train2017\000000000036.jpg: cannot reshape array of size 55 into shape (2)

Friends, I'm stuck in this problem. Can someone help me?

jq9898 commented 1 year ago

This error occurred when I ran the YOLOV7-pose algorithm.

train: Scanning 'F:\code\yolov7-pose\point\train2017' images and labels... 18 found, 0 missing, 0 empty, 18 corrupted: 100%|██████████| 18/18 [00:00<00:00, 1199.57it/s] train: New cache created: F:\code\yolov7-pose\point\train2017.cache Traceback (most recent call last): File "F:/code/yolov7-pose/train.py", line 562, in train(hyp, opt, device, tb_writer) File "F:/code/yolov7-pose/train.py", line 204, in train image_weights=opt.image_weights, quad=opt.quad, prefix=colorstr('train: '), kpt_label=kpt_label) File "F:\code\yolov7-pose\utils\datasets.py", line 74, in create_dataloader kpt_label=kpt_label) File "F:\code\yolov7-pose\utils\datasets.py", line 414, in init labels, shapes, self.segments = zip(*cache.values()) ValueError: not enough values to unpack (expected 3, got 0) train: WARNING: Ignoring corrupted image and/or label F:\code\yolov7-pose\point\images\train2017\000000000036.jpg: cannot reshape array of size 55 into shape (2)

Friends, I'm stuck in this problem. Can someone help me?

I am also troubled by this problem. Please let me know if there is a solution

aoqiangma commented 1 year ago

Ok, but I'm also a beginner. If any of us solve this problem, please tell each other the solution in the mailbox.

@.***

@.*** |

---- Replied Message ---- | From | @.> | | Date | 11/17/2022 11:03 | | To | @.> | | Cc | @.**@.> | | Subject | Re: [WongKinYiu/yolov7] YOLOV7-pose algorithm. (Issue #1110) |

This error occurred when I ran the YOLOV7-pose algorithm.

train: Scanning 'F:\code\yolov7-pose\point\train2017' images and labels... 18 found, 0 missing, 0 empty, 18 corrupted: 100%|██████████| 18/18 [00:00<00:00, 1199.57it/s] train: New cache created: F:\code\yolov7-pose\point\train2017.cache Traceback (most recent call last): File "F:/code/yolov7-pose/train.py", line 562, in train(hyp, opt, device, tb_writer) File "F:/code/yolov7-pose/train.py", line 204, in train image_weights=opt.image_weights, quad=opt.quad, prefix=colorstr('train: '), kpt_label=kpt_label) File "F:\code\yolov7-pose\utils\datasets.py", line 74, in create_dataloader kpt_label=kpt_label) File "F:\code\yolov7-pose\utils\datasets.py", line 414, in init labels, shapes, self.segments = zip(*cache.values()) ValueError: not enough values to unpack (expected 3, got 0) train: WARNING: Ignoring corrupted image and/or label F:\code\yolov7-pose\point\images\train2017\000000000036.jpg: cannot reshape array of size 55 into shape (2)

Friends, I'm stuck in this problem. Can someone help me?

I am also troubled by this problem. Please let me know if there is a solution

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

Liagos commented 1 year ago

Insert the --kpt-label argument and it will solve the issue

akashAD98 commented 1 year ago

how can i train on my custom pose classes?

Franksongsong commented 1 year ago

first delete train.cache and then run python train.py --kpt-label, it works well

schemaphicGopa commented 11 months ago

yes i delete train.cache and then run python train.py --kpt-label, it is not work how to solve

schemaphicGopa commented 11 months ago

i use --kpt-label but not solve problem

mohammedAljadd commented 9 months ago

I suspect the issue might be related to the annotation file.