mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feifei

pull/1806/head
feifei 2023-09-30 17:34:35 +08:00
parent e564c2203a
commit 8c287ed296
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mmpretrain/datasets/coco_retrieval.py
View File

@ -57,15 +57,15 @@ class COCORetrieval(BaseDataset):
>>> train_dataset
Dataset COCORetrieval
Number of samples: 414113
Annotation file: data/coco2014/annotations/captions_train2014.json
Prefix of images: data/coco2014/
Annotation file: /coco2014/annotations/captions_train2014.json
Prefix of images: /coco2014/
>>> from mmpretrain.datasets import COCORetrieval
>>> val_dataset = COCORetrieval(data_root='coco2014/')
>>> val_dataset
Dataset COCORetrieval
Number of samples: 202654
Annotation file: data/coco2014/annotations/captions_val2014.json
Prefix of images: data/coco2014/
Annotation file: /coco2014/annotations/captions_val2014.json
Prefix of images: /coco2014/
"""
def __init__(self,