Преглед изворни кода

add mixupimage to pdx.det.transforms

will-jl944 пре 4 година
родитељ
комит
ea2185947d
1 измењених фајлова са 2 додато и 2 уклоњено
  1. 2 2
      dygraph/paddlex/cv/transforms/det_transforms.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
+from .operators import Transform, Compose, ResizeByShort, Resize, RandomHorizontalFlip, Normalize, MixupImage
 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',
-    'Padding', 'RandomExpand', 'RandomCrop'
+    'MixupImage', 'Padding', 'RandomExpand', 'RandomCrop'
 ]