Bladeren bron

调整 efficientnet 默认m

RenLiqiang 7 maanden geleden
bovenliggende
commit
7b80da0bf0
1 gewijzigde bestanden met toevoegingen van 1 en 1 verwijderingen
  1. 1 1
      models/line_detect/line_net.py

+ 1 - 1
models/line_detect/line_net.py

@@ -338,7 +338,7 @@ _COMMON_META = {
 }
 
 
-def create_efficientnetv2_backbone(name='efficientnet_v2_l', pretrained=True):
+def create_efficientnetv2_backbone(name='efficientnet_v2_m', pretrained=True):
     # 加载EfficientNetV2模型
     if name == 'efficientnet_v2_s':
         weights = EfficientNet_V2_S_Weights.IMAGENET1K_V1 if pretrained else None