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

原创

已于 2023-09-04 17:50:43 修改 · 467 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#YOLO #机器学习 #人工智能 #pytorch

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

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: '),