yolo-v5代码数据读取部分理解

ydestspring

已于 2023-09-04 17:50:43 修改

阅读量445

点赞数 1

CC 4.0 BY-SA版权

文章标签： YOLO 机器学习人工智能 pytorch

于 2023-08-30 17:59:32 首次发布

本文链接：https://blog.youkuaiyun.com/ydestspring/article/details/132587684

1. train.py文件中，通过create_dataloader函数创建train_loader

train_loader, dataset = create_dataloader(train_path,
                                              imgsz,
                                              batch_size // WORLD_SIZE,
                                              gs,
                                              single_cls,
                                              hyp=hyp,
                                              augment=True,
                                              cache=None if opt.cache == 'val' else opt.cache,
                                              rect=opt.rect,
                                              rank=LOCAL_RANK,
                                              workers=workers,
                                              image_weights=opt.image_weights,
                                              quad=opt.quad,
                                              prefix=colorstr('train: '),