mirror of
https://github.com/modelscope/modelscope.git
synced 2026-09-01 19:49:03 +02:00
968e660235
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/8967432 * create ocr_detection task * replace c++ nms with python version * replace c++ decoder with python version * add requirements for ocr_detection
90 lines
2.6 KiB
Python
90 lines
2.6 KiB
Python
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
|
|
|
|
class Fields(object):
|
|
""" Names for different application fields
|
|
"""
|
|
# image = 'image'
|
|
# video = 'video'
|
|
cv = 'cv'
|
|
nlp = 'nlp'
|
|
audio = 'audio'
|
|
multi_modal = 'multi-modal'
|
|
|
|
|
|
class Tasks(object):
|
|
""" Names for tasks supported by modelscope.
|
|
|
|
Holds the standard task name to use for identifying different tasks.
|
|
This should be used to register models, pipelines, trainers.
|
|
"""
|
|
# vision tasks
|
|
image_to_text = 'image-to-text'
|
|
pose_estimation = 'pose-estimation'
|
|
image_classification = 'image-classification'
|
|
image_tagging = 'image-tagging'
|
|
object_detection = 'object-detection'
|
|
image_segmentation = 'image-segmentation'
|
|
image_editing = 'image-editing'
|
|
image_generation = 'image-generation'
|
|
image_matting = 'image-matting'
|
|
ocr_detection = 'ocr-detection'
|
|
|
|
# nlp tasks
|
|
word_segmentation = 'word-segmentation'
|
|
sentiment_analysis = 'sentiment-analysis'
|
|
sentence_similarity = 'sentence-similarity'
|
|
text_classification = 'text-classification'
|
|
relation_extraction = 'relation-extraction'
|
|
zero_shot = 'zero-shot'
|
|
translation = 'translation'
|
|
token_classification = 'token-classification'
|
|
conversational = 'conversational'
|
|
text_generation = 'text-generation'
|
|
table_question_answering = 'table-question-answering'
|
|
feature_extraction = 'feature-extraction'
|
|
sentence_similarity = 'sentence-similarity'
|
|
fill_mask = 'fill-mask '
|
|
summarization = 'summarization'
|
|
question_answering = 'question-answering'
|
|
|
|
# audio tasks
|
|
auto_speech_recognition = 'auto-speech-recognition'
|
|
text_to_speech = 'text-to-speech'
|
|
speech_signal_process = 'speech-signal-process'
|
|
|
|
# multi-modal tasks
|
|
image_captioning = 'image-captioning'
|
|
visual_grounding = 'visual-grounding'
|
|
text_to_image_synthesis = 'text-to-image-synthesis'
|
|
|
|
|
|
class InputFields(object):
|
|
""" Names for input data fields in the input data for pipelines
|
|
"""
|
|
img = 'img'
|
|
text = 'text'
|
|
audio = 'audio'
|
|
|
|
|
|
class Hubs(object):
|
|
""" Source from which an entity (such as a Dataset or Model) is stored
|
|
"""
|
|
modelscope = 'modelscope'
|
|
huggingface = 'huggingface'
|
|
|
|
|
|
class ModelFile(object):
|
|
CONFIGURATION = 'configuration.json'
|
|
README = 'README.md'
|
|
TF_SAVED_MODEL_FILE = 'saved_model.pb'
|
|
TF_GRAPH_FILE = 'tf_graph.pb'
|
|
TF_CHECKPOINT_FOLDER = 'tf_ckpts'
|
|
TF_CKPT_PREFIX = 'ckpt-'
|
|
TORCH_MODEL_FILE = 'pytorch_model.pt'
|
|
TORCH_MODEL_BIN_FILE = 'pytorch_model.bin'
|
|
|
|
|
|
TENSORFLOW = 'tensorflow'
|
|
PYTORCH = 'pytorch'
|