mmyolo.datasets.yolov5_voc 源代码
# Copyright (c) OpenMMLab. All rights reserved.
from mmdet.datasets import VOCDataset
from mmyolo.datasets.yolov5_coco import BatchShapePolicyDataset
from ..registry import DATASETS
[文档]@DATASETS.register_module()
class YOLOv5VOCDataset(BatchShapePolicyDataset, VOCDataset):
"""Dataset for YOLOv5 VOC Dataset.
We only add `BatchShapePolicy` function compared with VOCDataset. See
`mmyolo/datasets/utils.py#BatchShapePolicy` for details
"""
pass