Merge pull request #41 from zc277584121/main

add mpvit for image_embedding
towhee-io · Jun 29, 2022 · 65c52f9 · 65c52f9
2 parents f1294e4 + 67f741d
commit 65c52f9
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/image_embedding/description.yaml b/image_embedding/description.yaml
@@ -3,6 +3,15 @@ description: "Image embedding is a task that attempts to comprehend an entire im
 img-credit: "Convolutional Neural Network Layers and Architectures"
 img-ref: "https://www.researchgate.net/publication/333242381_Convolutional_Neural_Network_Layers_and_Architectures"
 models:
+    - mpvit:
+        model: "MPViT"
+        op-name: "image_embedding.mpvit"
+        op-link: "https://towhee.io/image-embedding/mpvit"
+        pretrained-models:
+            - "image_embedding.mpvit(model_name = 'mpvit_tiny')"
+            - "image_embedding.mpvit(model_name = 'mpvit_xsmall')"
+            - "image_embedding.mpvit(model_name = 'mpvit_small')"
+            - "image_embedding.mpvit(model_name = 'mpvit_base')"
     - data2vec:
         model: "data2vec"
         op-name: "image_embedding.data2vec"