Есть ли способ убрать серый фон с изображения и сохранить радугу? ⇐ Python

[Anonymous]

Радужная дифракционная картина


У меня есть это изображение, которое я пытаюсь обработать, и я хочу сохранить только видимые радуги и сделать фон черным вместо этого сероватого цвета. Возможно ли это?
Я попробовал кластеризацию, но обнаружил только цвета как счетчик:

Код: Выделить всё

import numpy as np
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans
from skimage.io import imread
from skimage.util import img_as_float

# Read image and convert to float [0,1]
img = img_as_float(imread('upscale.jpg'))

# Reshape image into N x 3 color pixels
pixels = img.reshape(-1, 3)

# Number of color clusters (you can adjust this)
k = 5

# Run k-means clustering on the colors
kmeans = KMeans(n_clusters=k, random_state=0, n_init=3)
idx = kmeans.fit_predict(pixels)
C = kmeans.cluster_centers_

# Reconstruct clustered image
clustered_img = C[idx].reshape(img.shape)

# Show original and clustered version
fig, axes = plt.subplots(1, 2, figsize=(12, 6))
axes[0].imshow(img)
axes[0].set_title('Original Image')
axes[0].axis('off')

axes[1].imshow(clustered_img)
axes[1].set_title(f'Color Clusters (k = {k})')
axes[1].axis('off')

plt.tight_layout()
plt.show()

Я хочу видеть только радужные пятна на черном фоне. Я также взял ПЗС-изображение радуги с дифракционной решетки.

Подробнее здесь: https://stackoverflow.com/questions/798 ... o-keep-the