Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-21 18:49:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix typo

Browse Source
This commit is contained in:
Yingda Chen
2022-05-20 06:24:18 +08:00
parent 5e469008fd
commit 3cbdcb1d3e
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
maas_lib/utils/constant.py
View File

@@ -9,7 +9,7 @@ class Fields(object):
cv = 'cv' cv = 'cv'
nlp = 'nlp' nlp = 'nlp'
audio = 'audio' audio = 'audio'
multi_modal = 'multi_modal' multi_modal = 'multi-modal'
class Tasks(object): class Tasks(object):
@@ -21,7 +21,7 @@ class Tasks(object):
# vision tasks # vision tasks
image_to_text = 'image-to-text' image_to_text = 'image-to-text'
pose_estimation = 'pose-estimation' pose_estimation = 'pose-estimation'
image_classfication = 'image-classification' image_classification = 'image-classification'
image_tagging = 'image-tagging' image_tagging = 'image-tagging'
object_detection = 'object-detection' object_detection = 'object-detection'
image_segmentation = 'image-segmentation' image_segmentation = 'image-segmentation'

14
tests/utils/test_registry.py
View File

@@ -25,13 +25,13 @@ class RegistryTest(unittest.TestCase):
def test_register_class_with_task(self): def test_register_class_with_task(self):
MODELS = Registry('models') MODELS = Registry('models')
@MODELS.register_module(Tasks.image_classfication, 'SwinT') @MODELS.register_module(Tasks.image_classification, 'SwinT')
class SwinTForCls(object): class SwinTForCls(object):
pass pass
self.assertTrue(Tasks.image_classfication in MODELS.modules) self.assertTrue(Tasks.image_classification in MODELS.modules)
self.assertTrue( self.assertTrue(
MODELS.get('SwinT', Tasks.image_classfication) is SwinTForCls) MODELS.get('SwinT', Tasks.image_classification) is SwinTForCls)
@MODELS.register_module(Tasks.sentiment_analysis, 'Bert') @MODELS.register_module(Tasks.sentiment_analysis, 'Bert')
class BertForSentimentAnalysis(object): class BertForSentimentAnalysis(object):
@@ -54,7 +54,7 @@ class RegistryTest(unittest.TestCase):
def test_list(self): def test_list(self):
MODELS = Registry('models') MODELS = Registry('models')
@MODELS.register_module(Tasks.image_classfication, 'SwinT') @MODELS.register_module(Tasks.image_classification, 'SwinT')
class SwinTForCls(object): class SwinTForCls(object):
pass pass
@@ -68,7 +68,7 @@ class RegistryTest(unittest.TestCase):
def test_build(self): def test_build(self):
MODELS = Registry('models') MODELS = Registry('models')
@MODELS.register_module(Tasks.image_classfication, 'SwinT') @MODELS.register_module(Tasks.image_classification, 'SwinT')
class SwinTForCls(object): class SwinTForCls(object):
pass pass
@@ -77,7 +77,7 @@ class RegistryTest(unittest.TestCase):
pass pass
cfg = dict(type='SwinT') cfg = dict(type='SwinT')
model = build_from_cfg(cfg, MODELS, Tasks.image_classfication) model = build_from_cfg(cfg, MODELS, Tasks.image_classification)
self.assertTrue(isinstance(model, SwinTForCls)) self.assertTrue(isinstance(model, SwinTForCls))
cfg = dict(type='Bert') cfg = dict(type='Bert')
@@ -86,7 +86,7 @@ class RegistryTest(unittest.TestCase):
with self.assertRaises(KeyError): with self.assertRaises(KeyError):
cfg = dict(type='Bert') cfg = dict(type='Bert')
model = build_from_cfg(cfg, MODELS, Tasks.image_classfication) model = build_from_cfg(cfg, MODELS, Tasks.image_classification)
if __name__ == '__main__': if __name__ == '__main__':