mmfewshot.detection.datasets.utils 源代码
# Copyright (c) OpenMMLab. All rights reserved.
import json
import numpy as np
[文档]class NumpyEncoder(json.JSONEncoder):
"""Save numpy array obj to json."""
[文档] def default(self, obj: object) -> object:
if isinstance(obj, np.ndarray):
return obj.tolist()
return json.JSONEncoder.default(self, obj)
[文档]def get_copy_dataset_type(dataset_type: str) -> str:
"""Return corresponding copy dataset type."""
if dataset_type in ['FewShotVOCDataset', 'FewShotVOCDefaultDataset']:
copy_dataset_type = 'FewShotVOCCopyDataset'
elif dataset_type in ['FewShotCocoDataset', 'FewShotCocoDefaultDataset']:
copy_dataset_type = 'FewShotCocoCopyDataset'
else:
raise TypeError(f'{dataset_type} '
f'not support copy data_infos operation.')
return copy_dataset_type