报错详情:
Traceback (most recent call last):
File "/home/zhangh/project1/yolov7-mask/train.py", line 617, in <module>
train(hyp, opt, device, tb_writer)
File "/home/zhangh/project1/yolov7-mask/train.py", line 246, in train
dataloader, dataset = create_dataloader(train_path, imgsz, batch_size, gs, opt,
File "/home/zhangh/project1/yolov7-mask/utils/datasets.py", line 69, in create_dataloader
dataset = LoadImagesAndLabels(path, imgsz, batch_size,
File "/home/zhangh/project1/yolov7-mask/utils/datasets.py", line 392, in __init__
cache, exists = torch.load(cache_path), True # load
File "/home/zhangh/anaconda3/envs/basebackup2/lib/python3.9/site-packages/torch/serialization.py", line 815, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/home/zhangh/anaconda3/envs/basebackup2/lib/python3.9/site-packages/torch/serialization.py", line 1033, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: STACK_GLOBAL requires str
检查是由于数据标签的cache的原因
因此删除.cache文件
最后再运行
完美解决!