Spaces:
Sleeping
Sleeping
File size: 361 Bytes
186701e |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
_base_ = 'yolov5_s-v61_syncbn_8xb16-300e_coco.py'
# fast means faster training speed,
# but less flexibility for multitasking
model = dict(
data_preprocessor=dict(
type='YOLOv5DetDataPreprocessor',
mean=[0., 0., 0.],
std=[255., 255., 255.],
bgr_to_rgb=True))
train_dataloader = dict(collate_fn=dict(type='yolov5_collate'))
|