Преобразование RGB в HSV в python - PullRequest
Новая
       90

Преобразование RGB в HSV в python

0 голосов
/ 27 мая 2020

Хотелось бы реализовать логи c за cv2.cvtColor(img,cv2.COLOR_BGR2HSV) с нуля. Я следовал формулам, выбранным на картинке ниже. Однако, когда я узнаю разницу между реализованной мной функцией и функцией opencv, вместо получения черного изображения я получил другой результат. Пожалуйста, помогите мне понять, что я сделал не так. Спасибо.

enter image description here

Реализация преобразования BGR2HSV. 

def convertBGRtoHSV(image):
###
### YOUR CODE HERE
###
  sample = (image * (1/255.0))
  B,G,R = cv2.split(sample)

  rows,cols,channels = sample.shape

  V = np.zeros(sample.shape[:2],dtype=np.float32)
  S = np.zeros(sample.shape[:2],dtype=np.float32)
  H = np.zeros(sample.shape[:2],dtype=np.float32)



  for i in range(rows):
      for j in range(cols):
          V[i,j] = max(B[i,j],G[i,j],R[i,j])
          Min_RGB = min(B[i,j],G[i,j],R[i,j])


          if V[i,j] != 0.0:
              S[i,j] = ((V[i,j] - Min_RGB) / V[i,j])
          else:
              S[i,j] = 0.0

          if V[i,j] == R[i,j]:
              H[i,j] = 60*(G[i,j] - B[i,j])/(V[i,j] - Min_RGB)
          elif V[i,j] == G[i,j]:
              H[i,j] = 120 + 60*(B[i,j] - R[i,j])/(V[i,j] - Min_RGB)
          elif V[i,j] == B[i,j]:
              H[i,j] = 240 + 60*(R[i,j] - G[i,j])/(V[i,j] - Min_RGB)

          if H[i,j] < 0:
              H[i,j] = H[i,j] + 360


  V = 255.0 * V
  S = 255.0 * S
  H = H/2
  hsv = np.round(cv2.merge((H,S,V)))
  return hsv.astype(np.int)

Результат выполнения приведенного выше кода приведен ниже. Разница должна быть нулем (черное изображение), но у меня другой результат.

enter image description here

1 Ответ

1 голос
/ 27 мая 2020

Вы сравниваете float32 V и float64 R, G, B V [i, j] == R [i, j] Это неверно. H равен нулю. Измените свой код: 

  V = np.zeros(sample.shape[:2],dtype=np.float64)
  S = np.zeros(sample.shape[:2],dtype=np.float64)
  H = np.zeros(sample.shape[:2],dtype=np.float64)
...