YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming), Quantization (MQBench) and Deployment (TensorRT, ncnn) Compression Tool Box.
994
stars
200
forks
source link
请问报这错啥原因啊,用的VisDrone数据集训练TPH-yolov5 #68
Open
KingTerminatora opened 2 years ago
Traceback (most recent call last): File "train.py", line 678, in
main(opt)
File "train.py", line 575, in main
train(opt.hyp, opt, device, callbacks)
File "train.py", line 230, in train
train_loader, dataset = create_dataloader(train_path, imgsz, batch_size // WORLD_SIZE, gs, single_cls,
File "H:\513work\FiberHomeWork\YOLOv5-Multibackbone-Compression-main\utils\datasets.py", line 99, in create_dataloader
dataset = LoadImagesAndLabels(path, imgsz, batch_size,
File "H:\513work\FiberHomeWork\YOLOv5-Multibackbone-Compression-main\utils\datasets.py", line 435, in init
labels, shapes, self.segments = zip(*cache.values())
ValueError: not enough values to unpack (expected 3, got 0)