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

add comments for metalearning

pull/1/head
zhangshilong 2021-05-16 19:24:15 +08:00
parent 314a2d1555
commit d4c08fe6d1
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mmfewshot/builders/dataset_builder.py
View File

@ -6,6 +6,9 @@ from mmdet.datasets.builder import build_dataset as build_det_dataset
def build_dataloader(dataset=None, task_type='mmdet', round_up=True, **kwargs): def build_dataloader(dataset=None, task_type='mmdet', round_up=True, **kwargs):
# TODO: identify how to bulid the dataloader via the type of dataset # TODO: identify how to bulid the dataloader via the type of dataset
# just an example
# if isinstance(dataset,base_meta_learning_dataset):
# data_loader=build_det_metalearning_dataloader(dataset=dataset, **kwargs)
if task_type == 'mmdet': if task_type == 'mmdet':
data_loader = build_det_dataloader(dataset=dataset, **kwargs) data_loader = build_det_dataloader(dataset=dataset, **kwargs)
elif task_type == 'mmcls': elif task_type == 'mmcls':
@ -25,3 +28,8 @@ def build_dataset(*args, task_type='mmdet', **kwargs):
else: else:
raise NotImplementedError raise NotImplementedError
return dataset return dataset
# TODO: check whether det and cls can use same dataloader for meta_learnig
def build_det_metalearning_dataloader():
pass