Use uint8 inputs in the example (#30)

albumentations-team · Jul 18, 2018 · 80b8526 · 80b8526
1 parent 6dcc995
commit 80b8526
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -48,8 +48,8 @@ def strong_aug(p=.5):
         HueSaturationValue(p=0.3),
     ], p=p)
 
-image = np.ones((300, 300))
-mask = np.ones((300, 300))
+image = np.ones((300, 300, 3), dtype=np.uint8)
+mask = np.ones((300, 300), dtype=np.uint8)
 whatever_data = "my name"
 augmentation = strong_aug(p=0.9)
 data = {"image": image, "mask": mask, "whatever_data": whatever_data, "additional": "hello"}

diff --git a/docs/examples.rst b/docs/examples.rst
@@ -44,8 +44,8 @@ Examples
            HueSaturationValue(p=0.3),
        ], p=p)
 
-   image = np.ones((300, 300))
-   mask = np.ones((300, 300))
+   image = np.ones((300, 300, 3), dtype=np.uint8)
+   mask = np.ones((300, 300), dtype=np.uint8)
    whatever_data = "my name"
    augmentation = strong_aug(p=0.9)
    data = {"image": image, "mask": mask, "whatever_data": whatever_data, "additional": "hello"}