Browse Source

fix dtype of imgaug

sunyanfang01 5 years ago
parent
commit
b6581a2645
1 changed files with 1 additions and 1 deletions
  1. 1 1
      paddlex/cv/transforms/imgaug_support.py

+ 1 - 1
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
     import imgaug.augmentables.bbs as bbs
 
 
     aug_im = im.astype('uint8')
     aug_im = im.astype('uint8')
-    aug_im = augmenter.augment(image=aug_im)
+    aug_im = augmenter.augment(image=aug_im).astype('float32')
     return aug_im
     return aug_im
 
 
     # TODO imgaug的标注处理逻辑与paddlex已存的transform存在部分差异
     # TODO imgaug的标注处理逻辑与paddlex已存的transform存在部分差异