mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-16 16:27:45 +01:00
210ab40c54
* del _datasets_server import in hf_dataset_util * fix streaming for youku-mplug and adopt latest datasets * fix download config copy * update ut * add youku in test_general_datasets * update UT for general dataset * adapt to datasets version: 2.19.0 or later * add assert for youku data UT * fix disable_tqdm in some functions for 2.19.0 or later * update get_module_with_script * set trust_remote_code is True in load_dataset_with_ctx * update print info * update requirements for datasets version restriction * fix _dataset_info * add pillow * update comments * update comment * reuse _download function in DataDownloadManager * remove unused code * update test_run_modelhub in Human3DAnimationTest * set datasets>=2.18.0
277 lines
11 KiB
Python
277 lines
11 KiB
Python
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
import hashlib
|
|
import os
|
|
import unittest
|
|
|
|
from modelscope.hub.snapshot_download import snapshot_download
|
|
from modelscope.models import Model
|
|
from modelscope.msdatasets import MsDataset
|
|
from modelscope.msdatasets.dataset_cls.custom_datasets.audio.asr_dataset import \
|
|
ASRDataset
|
|
from modelscope.preprocessors import TextClassificationTransformersPreprocessor
|
|
from modelscope.preprocessors.base import Preprocessor
|
|
from modelscope.utils.config import Config
|
|
from modelscope.utils.constant import (DEFAULT_DATASET_NAMESPACE, DownloadMode,
|
|
ModelFile)
|
|
from modelscope.utils.test_utils import require_tf, require_torch, test_level
|
|
|
|
|
|
class ImgPreprocessor(Preprocessor):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.path_field = kwargs.pop('image_path', 'image_path')
|
|
self.width = kwargs.pop('width', 'width')
|
|
self.height = kwargs.pop('height', 'width')
|
|
|
|
def __call__(self, data):
|
|
import cv2
|
|
image_path = data.get(self.path_field)
|
|
if not image_path:
|
|
return None
|
|
img = cv2.imread(image_path)
|
|
return {
|
|
'image':
|
|
cv2.resize(img,
|
|
(data.get(self.height, 128), data.get(self.width, 128)))
|
|
}
|
|
|
|
|
|
class GenLocalFile:
|
|
|
|
@staticmethod
|
|
def gen_mock_data() -> (str, str):
|
|
mock_data_list = [
|
|
'Title,Content,Label', 'mock title1,mock content1,mock label1',
|
|
'mock title2,mock content2,mock label2',
|
|
'mock title3,mock content3,mock label3'
|
|
]
|
|
|
|
mock_file_name = 'mock_file.csv'
|
|
md = hashlib.md5()
|
|
md.update('GenLocalFile.gen_mock_data.out_file_path'.encode('utf-8'))
|
|
mock_dir = os.path.join(os.getcwd(), md.hexdigest())
|
|
os.makedirs(mock_dir, exist_ok=True)
|
|
mock_relative_path = os.path.join(md.hexdigest(), mock_file_name)
|
|
with open(mock_relative_path, 'w') as f:
|
|
for line in mock_data_list:
|
|
f.write(line + '\n')
|
|
|
|
return mock_relative_path, md.hexdigest()
|
|
|
|
@staticmethod
|
|
def clear_mock_dir(mock_dir) -> None:
|
|
import shutil
|
|
shutil.rmtree(mock_dir)
|
|
|
|
|
|
class MsDatasetTest(unittest.TestCase):
|
|
|
|
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
|
|
def test_movie_scene_seg_toydata(self):
|
|
ms_ds_train = MsDataset.load('movie_scene_seg_toydata', split='train')
|
|
print(ms_ds_train._hf_ds.config_kwargs)
|
|
assert next(iter(ms_ds_train.config_kwargs['split_config'].values()))
|
|
assert next(iter(ms_ds_train))
|
|
|
|
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
|
|
def test_coco(self):
|
|
ms_ds_train = MsDataset.load(
|
|
'pets_small',
|
|
namespace=DEFAULT_DATASET_NAMESPACE,
|
|
download_mode=DownloadMode.FORCE_REDOWNLOAD,
|
|
split='train')
|
|
print(ms_ds_train.config_kwargs)
|
|
assert next(iter(ms_ds_train.config_kwargs['split_config'].values()))
|
|
|
|
@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
|
|
def test_ms_csv_basic(self):
|
|
ms_ds_train = MsDataset.load(
|
|
'clue', subset_name='afqmc',
|
|
split='train').to_hf_dataset().select(range(5))
|
|
print(next(iter(ms_ds_train)))
|
|
|
|
@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
|
|
def test_load_local_csv(self):
|
|
mock_relative_path, mock_dir_name = GenLocalFile.gen_mock_data()
|
|
# To test dataset_name in the form of `xxx/xxx.csv`
|
|
ds_from_single_file = MsDataset.load(mock_relative_path)
|
|
# To test dataset_name in the form of `xxx/`
|
|
ds_from_dir = MsDataset.load(mock_dir_name + '/')
|
|
|
|
GenLocalFile.clear_mock_dir(mock_dir_name)
|
|
ds_from_single_file_sample = next(iter(ds_from_single_file))
|
|
ds_from_dir_sample = next(iter(ds_from_dir))
|
|
|
|
print(ds_from_single_file_sample)
|
|
print(ds_from_dir_sample)
|
|
assert ds_from_single_file_sample
|
|
assert ds_from_dir_sample
|
|
|
|
@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
|
|
def test_ds_basic(self):
|
|
ms_ds_full = MsDataset.load(
|
|
'xcopa', subset_name='translation-et', namespace='damotest')
|
|
ms_ds = MsDataset.load(
|
|
'xcopa',
|
|
subset_name='translation-et',
|
|
namespace='damotest',
|
|
split='test')
|
|
print(next(iter(ms_ds_full['test'])))
|
|
print(next(iter(ms_ds)))
|
|
|
|
@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
|
|
@require_torch
|
|
def test_to_torch_dataset_text(self):
|
|
model_id = 'damo/nlp_structbert_sentence-similarity_chinese-tiny'
|
|
nlp_model = Model.from_pretrained(model_id)
|
|
preprocessor = TextClassificationTransformersPreprocessor(
|
|
nlp_model.model_dir,
|
|
first_sequence='premise',
|
|
second_sequence=None,
|
|
padding='max_length')
|
|
ms_ds_train = MsDataset.load(
|
|
'xcopa',
|
|
subset_name='translation-et',
|
|
namespace='damotest',
|
|
split='test')
|
|
pt_dataset = ms_ds_train.to_torch_dataset(preprocessors=preprocessor)
|
|
import torch
|
|
dataloader = torch.utils.data.DataLoader(pt_dataset, batch_size=5)
|
|
print(next(iter(dataloader)))
|
|
|
|
@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
|
|
@require_tf
|
|
def test_to_tf_dataset_text(self):
|
|
import tensorflow as tf
|
|
tf.compat.v1.enable_eager_execution()
|
|
model_id = 'damo/nlp_structbert_sentence-similarity_chinese-tiny'
|
|
nlp_model = Model.from_pretrained(model_id)
|
|
preprocessor = TextClassificationTransformersPreprocessor(
|
|
nlp_model.model_dir,
|
|
first_sequence='premise',
|
|
second_sequence=None)
|
|
ms_ds_train = MsDataset.load(
|
|
'xcopa',
|
|
subset_name='translation-et',
|
|
namespace='damotest',
|
|
split='test')
|
|
tf_dataset = ms_ds_train.to_tf_dataset(
|
|
batch_size=5,
|
|
shuffle=True,
|
|
preprocessors=preprocessor,
|
|
drop_remainder=True)
|
|
print(next(iter(tf_dataset)))
|
|
|
|
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
|
|
def test_to_dataset_asr(self):
|
|
ms_ds_asr = ASRDataset.load(
|
|
'speech_asr_aishell1_trainsets', namespace='speech_asr')
|
|
print(next(iter(ms_ds_asr['train'])))
|
|
|
|
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
|
|
@require_tf
|
|
def test_to_tf_dataset_img(self):
|
|
import tensorflow as tf
|
|
tf.compat.v1.enable_eager_execution()
|
|
ms_image_train = MsDataset.load(
|
|
'fixtures_image_utils', namespace='damotest', split='test')
|
|
tf_dataset = ms_image_train.to_tf_dataset(
|
|
batch_size=5,
|
|
shuffle=True,
|
|
preprocessors=ImgPreprocessor(image_path='file'),
|
|
drop_remainder=True,
|
|
)
|
|
print(next(iter(tf_dataset)))
|
|
|
|
@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
|
|
def test_streaming_load_uni_fold(self):
|
|
"""Test case for loading large scale datasets."""
|
|
dataset = MsDataset.load(
|
|
dataset_name='Uni-Fold-Data',
|
|
split='train',
|
|
use_streaming=True,
|
|
namespace='DPTech')
|
|
data_example = next(iter(dataset))
|
|
print(data_example)
|
|
assert data_example.values()
|
|
|
|
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
|
|
def test_streaming_load_afqmc(self):
|
|
"""To streaming-load afqmc dataset, which contains train/dev/validation data in meta-files."""
|
|
dataset = MsDataset.load('afqmc', split='test', use_streaming=True)
|
|
data_example = next(iter(dataset))
|
|
print(data_example)
|
|
assert data_example.values()
|
|
|
|
@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
|
|
def test_streaming_load_from_hf(self):
|
|
"""Use stream mode to load dataset from huggingface hub."""
|
|
from modelscope.utils.constant import Hubs
|
|
ds_train = MsDataset.load(
|
|
'glue',
|
|
subset_name='sst2',
|
|
split='train',
|
|
hub=Hubs.huggingface,
|
|
use_streaming=True)
|
|
data_example = next(iter(ds_train))
|
|
print(data_example)
|
|
assert data_example.values()
|
|
|
|
@unittest.skipUnless(test_level() >= 3, 'skip test in current test level')
|
|
def test_streaming_load_img_object(self):
|
|
"""Test case for iterating PIL object."""
|
|
from PIL.PngImagePlugin import PngImageFile
|
|
dataset = MsDataset.load(
|
|
dataset_name='SIDD',
|
|
subset_name='default',
|
|
namespace='huizheng',
|
|
split='train',
|
|
use_streaming=False)
|
|
data_example = next(iter(dataset))
|
|
print(data_example)
|
|
assert data_example.values()
|
|
|
|
@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
|
|
def test_to_ms_dataset(self):
|
|
"""Test case for converting huggingface dataset to `MsDataset` instance."""
|
|
from datasets.load import load_dataset
|
|
hf_dataset = load_dataset(
|
|
'AI-Lab-Makerere/beans', split='train', streaming=True)
|
|
ms_dataset = MsDataset.to_ms_dataset(hf_dataset)
|
|
data_example = next(iter(ms_dataset))
|
|
print(data_example)
|
|
assert data_example.values()
|
|
|
|
@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
|
|
def test_to_custom_dataset_movie_scene_toydata(self):
|
|
from modelscope.msdatasets.dataset_cls.custom_datasets.movie_scene_segmentation import \
|
|
MovieSceneSegmentationDataset
|
|
from modelscope.msdatasets.dataset_cls import ExternalDataset
|
|
|
|
model_id = 'damo/cv_resnet50-bert_video-scene-segmentation_movienet'
|
|
cache_path = snapshot_download(model_id)
|
|
config_path = os.path.join(cache_path, ModelFile.CONFIGURATION)
|
|
cfg = Config.from_file(config_path)
|
|
|
|
# ds_test.ds_instance got object 'MovieSceneSegmentationDataset' when the custom_cfg is not none.
|
|
ds_test_1 = MsDataset.load(
|
|
'modelscope/movie_scene_seg_toydata',
|
|
split='test',
|
|
custom_cfg=cfg,
|
|
test_mode=True)
|
|
assert ds_test_1.is_custom
|
|
assert isinstance(ds_test_1.ds_instance, MovieSceneSegmentationDataset)
|
|
|
|
# ds_test.ds_instance got object 'ExternalDataset' when the custom_cfg is none. (by default)
|
|
ds_test_2 = MsDataset.load(
|
|
'modelscope/movie_scene_seg_toydata',
|
|
split='test',
|
|
custom_cfg=None)
|
|
assert not ds_test_2.is_custom
|
|
assert isinstance(ds_test_2.ds_instance, ExternalDataset)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|