Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

RandomShadow input type wrong

See original GitHub issue

🐛 Bug

Weather transformation. For RandomRain, RandomSnow, RandomSunFlare the inputs are just numpy uint8. However, RandomShadow does not allow the same input format.

To Reproduce

albu_shadow = albu.RandomShadow(p=1, num_shadows_lower=1, num_shadows_upper=1, shadow_dimension=5, shadow_roi=(0, 0.5, 1, 1))

x_np = albu_shadow(image=x_np)['image']

Error:

TypeError: Expected Ptr<cv::UMat> for argument img

Expected behavior

It should take the same uint8 numpy array as input.

Environment

Albumentations version: 0.4.5
Python version: 3.7.6
OS (e.g., Linux): Linux
How you installed albumentations: pip

Issue Analytics

State:
Created 3 years ago
Reactions:1
Comments:15 (1 by maintainers)

Top GitHub Comments

2reactions

Dipetcommented, May 5, 2020

Very interesting. Problem how pytorch, numpy and OpenCV share memory. Until we find solution how to resolve problem you can use this fix to avoid problem:

x_np = albu_shadow(image=x_np.astype(np.uint8).copy())['image']

You need directly create numpy copy of image before provide it into albumentations transfromation

1reaction

Dipetcommented, Oct 6, 2021

Good point! Thank you for your comment. I reproduced that problem in the last version. I think we could resolve this problem by just adding np.ascontiguousarray for the image there https://github.com/albumentations-team/albumentations/blob/master/albumentations/core/transforms_interface.py#L97