浏览代码

add missing transforms to det_transforms.py

will-jl944 4 年之前
父节点
当前提交
d169f2472d
共有 2 个文件被更改,包括 4 次插入2 次删除
  1. 2 2
      dygraph/paddlex/cv/transforms/det_transforms.py
  2. 2 0
      dygraph/paddlex/det.py

+ 2 - 2
dygraph/paddlex/cv/transforms/det_transforms.py

@@ -17,14 +17,14 @@ function:
 """
 
 import numpy as np
-from .operators import Transform, Compose, ResizeByShort, Resize, RandomHorizontalFlip, Normalize, MixupImage
+from .operators import Transform, Compose, ResizeByShort, Resize, RandomHorizontalFlip, Normalize, MixupImage, RandomDistort
 from .operators import RandomExpand as dy_RandomExpand
 from .operators import RandomCrop as dy_RandomCrop
 from .functions import is_poly, expand_poly, expand_rle
 
 __all__ = [
     'Compose', 'ResizeByShort', 'Resize', 'RandomHorizontalFlip', 'Normalize',
-    'MixupImage', 'Padding', 'RandomExpand', 'RandomCrop'
+    'MixupImage', 'Padding', 'RandomExpand', 'RandomCrop', 'RandomDistort'
 ]
 
 

+ 2 - 0
dygraph/paddlex/det.py

@@ -156,6 +156,8 @@ class PPYOLO(cv.models.PPYOLO):
             nms_iou_threshold=0.45,
             train_random_shapes=None,
             input_channel=None):
+        if backbone == 'ResNet50_vd_ssld':
+            backbone = 'ResNet50_vd_dcn'
         if with_dcn_v2 is not None:
             logging.warning(
                 "`with_dcn_v2` is deprecated in PaddleX 2.0 and will not take effect. "