Ver Fonte

Merge pull request #899 from will-jl944/develop_jf

add mobilenet ssld models to cls.py
will-jl944 há 4 anos atrás
pai
commit
b5921970a0
1 ficheiros alterados com 2 adições e 0 exclusões
  1. 2 0
      dygraph/paddlex/cls.py

+ 2 - 0
dygraph/paddlex/cls.py

@@ -46,7 +46,9 @@ ResNet200_vd = cv.models.ResNet200_vd
 MobileNetV1 = cv.models.MobileNetV1
 MobileNetV2 = cv.models.MobileNetV2
 MobileNetV3_small = cv.models.MobileNetV3_small
+MobileNetV3_small_ssld = cv.models.MobileNetV3_small_ssld
 MobileNetV3_large = cv.models.MobileNetV3_large
+MobileNetV3_large_ssld = cv.models.MobileNetV3_large_ssld
 
 AlexNet = cv.models.AlexNet