Spaces:
Running
on
T4
Running
on
T4
_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')) | |