fix dtype of imgaug

b6581a26 · sunyanfang01 · f98427f3 · b6581a26
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddlex/cv/transforms/imgaug_support.py paddlex/cv/transforms/imgaug_support.py +1 -1

未找到文件。
--- a/paddlex/cv/transforms/imgaug_support.py
+++ b/paddlex/cv/transforms/imgaug_support.py
@@ -23,7 +23,7 @@ def execute_imgaug(augmenter, im, bboxes=None, polygons=None,
    import imgaug.augmentables.bbs as bbs

    aug_im = im.astype('uint8')
-    aug_im = augmenter.augment(image=aug_im)
+    aug_im = augmenter.augment(image=aug_im).astype('float32')
    return aug_im

    # TODO imgaug的标注处理逻辑与paddlex已存的transform存在部分差异