Browse Source

Merge pull request #792 from will-jl944/develop_jf

PaddleClas updated (add __init__.py)
FlyingQianMM 4 years ago
parent
commit
8373586423

+ 1 - 1
dygraph/PaddleClas

@@ -1 +1 @@
-Subproject commit c471929c18969273936c8b8893e1a4d0645fed7b
+Subproject commit 58a788db127c2692622a9279c1e46616ec2ce794

+ 6 - 0
dygraph/paddlex/cv/transforms/cls_transforms.py

@@ -23,6 +23,12 @@ from PIL import Image
 from .operators import Transform, Compose, RandomHorizontalFlip, RandomVerticalFlip, Normalize, \
     ResizeByShort, CenterCrop, RandomDistort, ArrangeClassifier
 
+__all__ = [
+    'Compose', 'RandomHorizontalFlip', 'RandomVerticalFlip', 'Normalize',
+    'ResizeByShort', 'CenterCrop', 'RandomDistort', 'ArrangeClassifier',
+    'RandomCrop', 'RandomRotate', 'ComposedClsTransforms'
+]
+
 
 class RandomCrop(Transform):
     """对图像进行随机剪裁,模型训练时的数据增强操作。

+ 5 - 0
dygraph/paddlex/cv/transforms/det_transforms.py

@@ -22,6 +22,11 @@ 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'
+]
+
 
 class Padding(Transform):
     """1.将图像的长和宽padding至coarsest_stride的倍数。如输入图像为[300, 640],

+ 8 - 0
dygraph/paddlex/cv/transforms/seg_transforms.py

@@ -23,6 +23,14 @@ from .operators import Transform, Compose, RandomHorizontalFlip, RandomVerticalF
     ResizeByShort, Normalize, RandomDistort, ArrangeSegmenter
 from .operators import Padding as dy_Padding
 
+__all__ = [
+    'Compose', 'RandomHorizontalFlip', 'RandomVerticalFlip', 'Resize',
+    'ResizeByShort', 'Normalize', 'RandomDistort', 'ArrangeSegmenter',
+    'ResizeByLong', 'ResizeRangeScaling', 'ResizeStepScaling', 'Padding',
+    'RandomPaddingCrop', 'RandomBlur', 'RandomRotate', 'RandomScaleAspect',
+    'Clip', 'ComposedSegTransforms'
+]
+
 
 class ResizeByLong(Transform):
     """对图像长边resize到固定值,短边按比例进行缩放。当存在标注图像时,则同步进行处理。