Skip to content

S
seminar-breakout
Commit f0ee73fc authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

use same dtype for const background (fix #408)

parent 2615205e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/dataflow/imgaug/paste.py
View file @ f0ee73fc
...@@ -49,7 +49,7 @@ class ConstantBackgroundFiller(BackgroundFiller): ...@@ -49,7 +49,7 @@ class ConstantBackgroundFiller(BackgroundFiller):
return_shape = background_shape + (img.shape[2],) return_shape = background_shape + (img.shape[2],)
else: else:
return_shape = background_shape return_shape = background_shape
return np.zeros(return_shape) + self.value return np.zeros(return_shape, dtype=img.dtype) + self.value
class CenterPaste(ImageAugmentor): class CenterPaste(ImageAugmentor):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment