File size: 531 Bytes
186701e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
# Copyright (c) Tencent Inc. All rights reserved.
from mmdet.datasets import Objects365V1Dataset
from mmyolo.registry import DATASETS
from .utils import RobustBatchShapePolicyDataset
@DATASETS.register_module()
class YOLOv5Objects365V1Dataset(RobustBatchShapePolicyDataset,
Objects365V1Dataset):
"""Dataset for YOLOv5 VOC Dataset.
We only add `BatchShapePolicy` function compared with Objects365V1Dataset.
See `mmyolo/datasets/utils.py#BatchShapePolicy` for details
"""
pass
|