Переглянути джерело

Merge pull request #899 from will-jl944/develop_jf

add mobilenet ssld models to cls.py
will-jl944 4 роки тому
батько
коміт
b5921970a0
1 змінених файлів з 2 додано та 0 видалено
  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