Как диагностировать и исправить assertion failed в OpenCV Python

Расширенные примеры работы с ошибкой assertion

Пример 1: Обработка видеофайла с контролем кадров

При чтении видео кадры могут заканчиваться, и попытка обработать None вызовет assertion. Ниже приведён безопасный цикл с проверкой.

import cv2

cap = cv2.VideoCapture('video.mp4')
if not cap.isOpened():
    print('Не удалось открыть видео')
    exit()

while True:
    ret, frame = cap.read()
    if not ret:
        print('Конец видео или ошибка чтения')
        break
    # Проверка, что frame не None
    assert frame is not None, 'Получен пустой кадр'
    # Обработка кадра (например, изменение размера)
    small = cv2.resize(frame, (640, 480))
    cv2.imshow('Frame', small)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

Результат: видео проигрывается до конца или до нажатия 'q'. В случае пустого кадра цикл прервётся без ошибки.

Пример 2: Маскирование с использованием bitwise_and при несовпадающих размерах

Часто требуется применить маску к изображению. Ошибка возникает, если маска меньше или больше исходного изображения.

img = cv2.imread('scene.jpg')
mask = cv2.imread('mask.png', cv2.IMREAD_GRAYSCALE)

# Приведение маски к размеру изображения
if mask.shape != img.shape[:2]:
    mask = cv2.resize(mask, (img.shape[1], img.shape[0]), interpolation=cv2.INTER_NEAREST)

# Применение маски (mask должна быть двумерной, а img трёхмерной)
masked = cv2.bitwise_and(img, img, mask=mask)
cv2.imshow('Masked', masked)

Результат: корректное наложение маски. Если маска имеет 3 канала, возникает ошибка - требуется одноканальная маска.

Пример 3: Имитация ошибки и её отладка через assert собственный

Полезно самостоятельно выбрасывать исключение с понятным сообщением.

def safe_resize(img, width, height):
    assert img is not None, 'Изображение не загружено'
    assert img.shape[0] > 0 and img.shape[1] > 0, 'Размеры изображения некорректны'
    return cv2.resize(img, (width, height))

img = cv2.imread('nonexistent.jpg')
try:
    resized = safe_resize(img, 400, 300)
except AssertionError as e:
    print(f'Ошибка: {e}')

Вывод: Ошибка: Изображение не загружено

Пример 4: Использование cv2.imdecode для чтения из байтов (кириллица в пути)

Когда путь содержит нелатинские символы, cv2.imread возвращает None. Решение – чтение через numpy.fromfile.

import numpy as np

path = 'C:/Users/Пользователь/image.jpg'
with open(path, 'rb') as f:
    file_bytes = np.asarray(bytearray(f.read()), dtype=np.uint8)
img = cv2.imdecode(file_bytes, cv2.IMREAD_COLOR)
if img is None:
    print('Ошибка декодирования')
else:
    cv2.imshow('Image', img)

Результат: изображение отображается независимо от кодировки пути.

Пример 5: Проверка количества каналов перед cv2.calcHist

Функция cv2.calcHist ожидает список изображений, каждое из которых должно быть одноканальным. Если передать цветное изображение, возникнет assertion.

img = cv2.imread('color.jpg')
# Разделение на каналы
channels = cv2.split(img)
hist = []
for i, ch in enumerate(channels):
    hist.append(cv2.calcHist([ch], [0], None, [256], [0,256]))
    print(f'Гистограмма канала {i} вычислена')

Результат: вывод гистограмм для каждого канала без ошибок.