File size: 935 Bytes
d3dbf03 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
# Copyright (c) OpenMMLab. All rights reserved.
from typing import Dict
from mmcv.transforms import BaseTransform
from mmaction.registry import TRANSFORMS
@TRANSFORMS.register_module()
class CLIPTokenize(BaseTransform):
"""Tokenize text and convert to tensor."""
def transform(self, results: Dict) -> Dict:
"""The transform function of :class:`CLIPTokenize`.
Args:
results (dict): The result dict.
Returns:
dict: The result dict.
"""
try:
import clip
except ImportError:
raise ImportError('Please run `pip install '
'git+https://github.com/openai/CLIP.git` '
'to install clip first. ')
text = results['text']
text_tokenized = clip.tokenize(text)[0]
results['text'] = text_tokenized
return results
|