Add dependency to opencv-contrib-python and opencv-contrib-python-headless

[tastelessjolt]

🐛 Add dependency to opencv-contrib-python and opencv-contrib-python-headless

Not exactly a bug. If other parts of the code use extra modules provided by the contrib module, but also want to use albumentations.

To Reproduce

Steps to reproduce the behavior:

1. Install albumentations in an environment where cv2 is provided by opencv-contrib-python or opencv-contrib-python-headless

Expected behavior

Can take the opencv-contrib-python or opencv-contrib-python-headless as dependencies while installing

Environment

• Albumentations version (e.g., 0.1.8): 0.4.6
• Python version (e.g., 3.7): 3.7
• OS (e.g., Linux): Linux
• How you installed albumentations (conda, pip, source): pip
• Any other relevant information:

Additional context

Reference for what these packages provide: https://github.com/skvark/opencv-python#installation-and-usage

Maybe you can add these entries in here.

albumentations/setup.py

Line 11 in 7f099ad

CHOOSE_INSTALL_REQUIRES = [("opencv-python>=4.1.1", "opencv-python-headless>=4.1.1")]

[agchang-cgl]

+1, this makes it a pain to use in a production system that needs the opencv-contrib-* dependency.

[Dipet]

Fixed in #837