Browse Source
add uniformer
Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
main
gexy5
3 years ago
1 changed files with
1 additions and
1 deletions
-
uniformer.py
|
|
@ -16,7 +16,7 @@ from towhee.models.uniformer.uniformer import create_model |
|
|
log = logging.getLogger() |
|
|
log = logging.getLogger() |
|
|
|
|
|
|
|
|
@register(output_schema=['labels', 'scores', 'features']) |
|
|
@register(output_schema=['labels', 'scores', 'features']) |
|
|
class Tsm(NNOperator): |
|
|
|
|
|
|
|
|
class Uniformer(NNOperator): |
|
|
""" |
|
|
""" |
|
|
Generate a list of class labels given a video input data. |
|
|
Generate a list of class labels given a video input data. |
|
|
Default labels are from [Kinetics400 Dataset](https://deepmind.com/research/open-source/kinetics). |
|
|
Default labels are from [Kinetics400 Dataset](https://deepmind.com/research/open-source/kinetics). |
|
|
|
