Python CV ecosystem (OpenCV, PIL, torchvision) in Computer Vision

Introduction

Python CV ecosystem helps you work with images and videos easily. It lets you read, edit, and analyze pictures for projects like face detection or photo filters.

You want to open and show images in your program.

You need to apply filters or resize photos for a project.

You want to prepare images for a machine learning model.

You want to detect objects or faces in pictures or videos.

You want to convert images between different formats.

Syntax

Computer Vision

import cv2
from PIL import Image
import torchvision.transforms as transforms

OpenCV (cv2) is great for fast image and video processing.

PIL (Pillow) is simple for opening and editing images.

torchvision helps prepare images for deep learning models.

Examples

OpenCV reads and shows an image in a window.

Computer Vision

import cv2
img = cv2.imread('photo.jpg')
cv2.imshow('Image', img)
cv2.waitKey(0)
cv2.destroyAllWindows()

PIL opens an image, resizes it, and displays it.

Computer Vision

from PIL import Image
img = Image.open('photo.jpg')
img = img.resize((100, 100))
img.show()

torchvision transforms resize an image and convert it to a tensor for ML models.

Computer Vision

from PIL import Image
import torchvision.transforms as transforms
img = Image.open('photo.jpg')
transform = transforms.Compose([
    transforms.Resize((128, 128)),
    transforms.ToTensor()
])
tensor_img = transform(img)

Sample Model

This program shows how to open an image with PIL, convert it to OpenCV format, resize it, convert back to PIL, and finally transform it to a tensor using torchvision. It prints the size and shape at each step.

Computer Vision

import cv2
import numpy as np
from PIL import Image
import torchvision.transforms as transforms

# Open image with PIL
img_pil = Image.open('sample.jpg')
print(f'PIL image size: {img_pil.size}')

# Convert PIL image to OpenCV format
img_cv = cv2.cvtColor(np.array(img_pil), cv2.COLOR_RGB2BGR)
print(f'OpenCV image shape: {img_cv.shape}')

# Resize image using OpenCV
img_cv_resized = cv2.resize(img_cv, (64, 64))
print(f'Resized OpenCV image shape: {img_cv_resized.shape}')

# Convert back to PIL
img_pil_resized = Image.fromarray(cv2.cvtColor(img_cv_resized, cv2.COLOR_BGR2RGB))

# Use torchvision to transform image to tensor
transform = transforms.ToTensor()
tensor_img = transform(img_pil_resized)
print(f'Tensor shape: {tensor_img.shape}')

Important Notes

OpenCV uses BGR color order, while PIL uses RGB. Remember to convert colors when switching.

torchvision transforms are useful to prepare images for deep learning models.

Always check image shapes and sizes after transformations to avoid errors.

Summary

OpenCV, PIL, and torchvision are key tools for working with images in Python.

Use OpenCV for fast image/video processing, PIL for easy image editing, and torchvision for ML image preparation.

Converting between these libraries is common and requires attention to color formats and shapes.

Practice

1. Which Python library is best known for fast image and video processing tasks?

easy

A. PIL (Pillow)

B. OpenCV

C. torchvision

D. matplotlib

Solution

1. Step 1: Understand library purposes

   OpenCV is designed for fast image and video processing, widely used in computer vision.

2. Step 2: Compare with other libraries

   PIL is mainly for image editing, torchvision is for ML image datasets, matplotlib is for plotting.

3. Final Answer:

   OpenCV -> Option B

4. Quick Check:

   Fast image/video processing = OpenCV [OK]

Hint: OpenCV = fast image/video tasks, PIL = editing, torchvision = ML prep [OK]

Common Mistakes:

• Confusing PIL as the fastest for video processing
• Thinking torchvision handles video processing
• Assuming matplotlib is for image processing

2. Which of the following is the correct way to read an image using OpenCV in Python?

easy

A. img = cv2.imread('image.jpg')

B. img = Image.open('image.jpg')

C. img = torchvision.io.read_image('image.jpg')

D. img = plt.imread('image.jpg')

Solution

1. Step 1: Identify OpenCV image reading syntax

   OpenCV uses cv2.imread() to load images from files.

2. Step 2: Differentiate from other libraries

   PIL uses Image.open(), torchvision uses torchvision.io.read_image(), matplotlib uses plt.imread().

3. Final Answer:

   img = cv2.imread('image.jpg') -> Option A

4. Quick Check:

   OpenCV image read = cv2.imread() [OK]

Hint: OpenCV reads images with cv2.imread() [OK]

Common Mistakes:

• Using Image.open() which is from PIL, not OpenCV
• Using plt.imread() which is for plotting, not OpenCV
• Confusing torchvision's read_image with OpenCV

3. What will be the shape and color format of the image loaded by this OpenCV code?

import cv2
img = cv2.imread('image.jpg')
print(img.shape)

medium

A. (height, width, 3) with RGB color order

B. (width, height, 3) with BGR color order

C. (width, height, 3) with RGB color order

D. (height, width, 3) with BGR color order

Solution

1. Step 1: Understand OpenCV image shape

   OpenCV loads images as NumPy arrays with shape (height, width, channels).

2. Step 2: Know OpenCV color format

   OpenCV uses BGR color order by default, not RGB.

3. Final Answer:

   (height, width, 3) with BGR color order -> Option D

4. Quick Check:

   OpenCV shape = (H, W, 3), color = BGR [OK]

Hint: OpenCV images: shape (H,W,3), color BGR [OK]

Common Mistakes:

• Assuming RGB color order instead of BGR
• Swapping width and height in shape
• Thinking OpenCV loads grayscale by default

4. This code tries to convert a PIL image to a NumPy array for OpenCV processing but causes an error:

from PIL import Image
import numpy as np
img_pil = Image.open('image.jpg')
img_cv = np.array(img_pil)

What is the likely cause and fix?

medium

A. PIL image must be converted to grayscale first

B. Color channels are in wrong order; convert RGB to BGR after np.array()

C. Use img_pil.convert('RGB') before np.array() to ensure 3 channels

D. No error; code works fine as is

Solution

1. Step 1: Identify PIL image mode issue

   PIL images may not be in RGB mode by default; could be 'P' or 'L' mode causing np.array to have unexpected shape.

2. Step 2: Fix by converting to RGB mode

   Use img_pil.convert('RGB') to ensure 3 color channels before converting to NumPy array.

3. Final Answer:

   Use img_pil.convert('RGB') before np.array() to ensure 3 channels -> Option C

4. Quick Check:

   PIL to NumPy needs RGB mode [OK]

Hint: Convert PIL image to RGB before np.array() [OK]

Common Mistakes:

• Assuming np.array always works without convert()
• Ignoring color channel order differences
• Trying to convert to grayscale unnecessarily

5. You want to prepare an image for a PyTorch model using torchvision transforms. Which sequence correctly converts a PIL image to a tensor normalized for pretrained models?

hard

A. Use torchvision.transforms.ToTensor() then torchvision.transforms.Normalize(mean, std)

B. Use cv2.imread() then convert to tensor manually

C. Use PIL.Image.open() then convert to NumPy array and normalize manually

D. Use torchvision.transforms.Normalize() only on PIL image

Solution

1. Step 1: Understand torchvision transform pipeline

   To prepare images for PyTorch models, convert PIL image to tensor with ToTensor(), which scales pixels to [0,1].

2. Step 2: Normalize tensor with mean and std

   Use Normalize() with pretrained model's mean and std to standardize input.

3. Final Answer:

   Use torchvision.transforms.ToTensor() then torchvision.transforms.Normalize(mean, std) -> Option A

4. Quick Check:

   ToTensor + Normalize = correct PyTorch prep [OK]

Hint: Use ToTensor() then Normalize() for PyTorch image prep [OK]

Common Mistakes:

• Trying to normalize PIL images directly
• Using OpenCV images without conversion
• Skipping normalization step