search

iSEE-Laboratory / DiffUIR

The official implementation of the paper of CVPR2024: Selective Hourglass Mapping for Universal Image Restoration Based on Diffusion Model
MIT License
103 stars 7 forks source link

name 'cv2equalizeHist' is not defined #4

Closed 15735347877 closed 7 months ago

15735347877 commented 7 months ago

"Hello, may I ask if the code src/visualization-Line1279-1281 where input_image = cv2equalizeHist(input_image) is used for testing low-light images, is the function cv2equalizeHist written by the author? If it is a custom function, it is not present in the code; if not, calling cv2.equalizeHist can only perform histogram equalization on grayscale images? This piece of code will cause an error. I look forward to the author's reply." bb0593c5d0bc24df70e0d02c2be4241

zhengdian1 commented 7 months ago

Thank you for pointing out. We have updated the visualization.py and added the cv2equalizeHist function

15735347877 commented 7 months ago

Oh,thanks for your answer.