Merge pull request #20 from jaelgu/main

Add image_embedding.swag
towhee-io · Jun 16, 2022 · 64e6e09 · 64e6e09
2 parents 6544ef6 + 1fc0d3f
commit 64e6e09
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/image_embedding/description.yaml b/image_embedding/description.yaml
@@ -3,6 +3,17 @@ 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:
+    - swag:
+        model: "SWAG"
+        op-name: "image_embedding.swag"
+        op-link: "https://towhee.io/image-embedding/swag"
+        pretrained-models:
+            - "image_embedding.swag(model_name = 'vit_b16_in1k')"
+            - "image_embedding.swag(model_name = 'vit_l16_in1k')"
+            - "image_embedding.swag(model_name = 'vit_h14_in1k')"
+            - "image_embedding.swag(model_name = 'regnety_16gf_in1k')"
+            - "image_embedding.swag(model_name = 'regnety_32gf_in1k')"
+            - "image_embedding.swag(model_name = 'regnety_128gf_in1k')"
     - vgg:
         model: "VGG"
         op-name: "image_embedding.timm"
@@ -15,7 +26,7 @@ models:
             - "image_embedding.timm(model_name = 'vgg16')"
             - "image_embedding.timm(model_name = 'vgg16_bn')"
             - "image_embedding.timm(model_name = 'vgg19')"
-            - "image_embedding.timm(model_name = 'vgg19_bn')" 
+            - "image_embedding.timm(model_name = 'vgg19_bn')"
     - resnet:
         model: "ResNet"
         op-name: "image_embedding.timm"
@@ -310,7 +321,7 @@ models:
             - "image_embedding.timm(model_name = 'beit_large_patch16_224')"
             - "image_embedding.timm(model_name = 'beit_large_patch16_224_in22k')"
             - "image_embedding.timm(model_name = 'beit_large_patch16_384')"
-            - "image_embedding.timm(model_name = 'beit_large_patch16_512')" 
+            - "image_embedding.timm(model_name = 'beit_large_patch16_512')"
     - swintransformer:
         model: "Swin Transformer"
         op-name: "image_embedding.timm"
@@ -811,7 +822,7 @@ models:
             - "image_embedding.timm(model_name = 'tresnet_m_448')"
             - "image_embedding.timm(model_name = 'tresnet_m_miil_in21k')"
             - "image_embedding.timm(model_name = 'tresnet_xl')"
-            - "image_embedding.timm(model_name = 'tresnet_xl_448')" 
+            - "image_embedding.timm(model_name = 'tresnet_xl_448')"
     - twins:
         model: "Twins"
         op-name: "image_embedding.timm"