mirror of https://github.com/open-mmlab/mmyolo.git
13 lines
361 B
Python
13 lines
361 B
Python
_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'))
|