Проблема при сохранении PNG в JPG в OpenCV

Question

Я запускаю этот кусок кода и получаю неправильный результат:

        #saving image into a white bg
        img = cv2.imread(dir_img + id, cv2.IMREAD_UNCHANGED)
        img = cv2.cvtColor(img, cv2.COLOR_BGRA2BGR)
        print(img.shape)
        cv2.imwrite(dir_img + id, img, [int(cv2.IMWRITE_JPEG_QUALITY), 100])

Исходный файл представляет собой png с прозрачнымфон.Я не знаю почему, но это экономит с этим серым рисунком за горлышком бутылки.

Оригинальный файл:

Answer 1

Как уже упоминалось в комментариях, простое удаление альфа-канала не приводит к удалению фона в этом случае, потому что у канала BGR есть артефакт, который вы пытаетесь удалить, как показано ниже, когда вы строите только график каналов B, G или R .

А ваш альфа-канал выглядит так

Чтобы достичь того, что вам нужно, вам нужно применить некоторую матричную математику, чтобы получить свой результат. Я прикрепил код здесь

import cv2
import matplotlib.pyplot as plt

img_path = r"path/to/image"

#saving image into a white bg
img = cv2.imread(img_path, cv2.IMREAD_UNCHANGED)
plt.imshow(img)
plt.show()
b,g,r, a = cv2.split(img)
print(img.shape)

new_img  = cv2.merge((b, g, r))
not_a = cv2.bitwise_not(a)
not_a = cv2.cvtColor(not_a, cv2.COLOR_GRAY2BGR)
plt.imshow(not_a)
plt.show()
new_img = cv2.bitwise_and(new_img,new_img,mask = a)
new_img = cv2.add(new_img, not_a)

cv2.imwrite(output_dir, new_img)
plt.imshow(new_img)
print(new_img.shape)
plt.show()

В результате получается изображение с размерами (1200, 1200, 3)