search

hukaixuan19970627 / yolov5_obb

yolov5 + csl_label.(Oriented Object Detection)（Rotation Detection）（Rotated BBox）基于yolov5的旋转目标检测
GNU General Public License v3.0
1.84k stars 428 forks source link

训练模型时出现问题，报错找不到中心文件 #451

Open 2597883929 opened 2 years ago

2597883929 commented 2 years ago

git上拉下项目，过完install.md，用预训练模型可以运行 python val.py --task 'val' --device 0 --save-json --batch-size 2 --data 'data/yolov5obb_demo_split.yaml' --name 'obb_demo_split' 但是自己训练的时候，使用 python train.py --device 3 报错RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory 使用项目自带的数据集，python train.py --data 'data/yolov5obb_demo.yaml' --epochs 10 --batch-size 1 --img 1024 --device 0 出现相同报错 PytorchStreamReader failed reading zip archive: failed finding central directory

WangGS102 commented 1 year ago

你好，请问解决了吗？？我运行这个demo的时候python train.py --weights "weights/yolov5x.pt" --data "data/yolov5obb_demo.yaml" --hyp "data/hyps/obb/hyp.finetune_dota.yaml" --epochs 10 --batch-size 1 --img 1024 --device 0 也出现了这个错误RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory