1. 程式人生 > >區域性邊緣保持濾波(LEP)高動態範圍影象HDR壓縮 matlab程式(二)

區域性邊緣保持濾波(LEP)高動態範圍影象HDR壓縮 matlab程式(二)

上一篇部落格給出了論文"Gu B, Li W, Zhu M, et al. Local edge-preserving multiscale decomposition for high dynamic range image tone mapping[J]. Image Processing, IEEE Transactions on, 2013, 22(1): 70-79.

下面為高動態範圍壓縮程式:

  1. clear all;  
  2. HDR = hdrread('..\HDR Images\AhwahneeGreatLounge_small.hdr');  
  3. % HDR = hdrread('..\HDR Images\AtriumMorning.hdr'
    );  
  4. % HDR = hdrread('..\HDR Images\belgium.hdr');  
  5. % HDR = hdrread('..\HDR Images\cadik-desk02_mid.hdr');  
  6. % HDR = hdrread('..\HDR Images\designCenter.hdr');  
  7. % HDR = hdrread('..\HDR Images\desk.hdr');  
  8. % HDR = hdrread('..\HDR Images\doll.hdr');  
  9. % HDR = hdrread('..\HDR Images\groveD.hdr');  
  10. % HDR = hdrread('..\HDR Images\HancockKitchenInside_small.hdr'
    );  
  11. % HDR = hdrread('..\HDR Images\memorial.hdr');  
  12. % HDR = hdrread('..\HDR Images\orion_correct_small.hdr');  
  13. % HDR = hdrread('..\HDR Images\paul_bunyan_small.hdr');  
  14. % HDR = hdrread('..\HDR Images\pillarsB_small.hdr');  
  15. % HDR = hdrread('..\HDR Images\snowman.hdr');  
  16. % HDR = hdrread('..\HDR Images\tinterna_small.hdr'
    );  
  17. % HDR = hdrread('..\HDR Images\vinesunset.hdr');  
  18. % HDR = hdrread('..\HDR Images\yosemite_small.hdr');  
  19. L_in=(1/3)*(HDR(:,:,1)+HDR(:,:,2)+HDR(:,:,3));  
  20. L=log(L_in*1e6+1);  
  21. L=L/max(max(L));  
  22. % L=mat2gray(L);  
  23. alpha=0.1;  
  24. beta=1;  
  25. r=2;  
  26. nLevel = 3;  
  27. B = cell(1, nLevel);  
  28. D = cell(1, nLevel);  
  29. D_compression=cell(1, nLevel);  
  30. B{nLevel}=L;  
  31. for j = nLevel:-1:2  
  32.     B{j-1}=LocalWls_HDR(B{j}, alpha, beta, r);  
  33.     D{j}=B{j}-B{j-1};  
  34.     r=20;  
  35. end  
  36. B0=mean(mean(B{1}))*ones(size(L));  
  37. D{1}=B{1}-B0;  
  38. for j = nLevel:-1:1  
  39.     D_compression{j}=(2/pi)*atan(20*D{j});  
  40.     D_compression{j}=mat2gray(D_compression{j});  
  41. end  
  42. L_out=D_compression{1}*0.5+D_compression{2}+D_compression{3};  
  43. % Rmax_clip = prctile(L_out(:),99);  
  44. % Rmin_clip = prctile(L_out(:),1);  
  45. % DR_clip = Rmax_clip/Rmin_clip;  
  46. % exponent = log(100)/log(DR_clip);  
  47. % L_out = max(0,L_out/Rmax_clip) .^ exponent;  
  48. Rmax_clip = prctile(L_out(:),99.5);  
  49. Rmin_clip = prctile(L_out(:),0.5);  
  50. L_out(L_out>Rmax_clip)=Rmax_clip;  
  51. L_out(L_out<Rmin_clip)=Rmin_clip;  
  52. L_out=mat2gray(L_out);  
  53. %L_out=L_out/max(max(L_out));  
  54. out(:,:,1)=((HDR(:,:,1)./L_in).^0.6).*L_out;  
  55. out(:,:,2)=((HDR(:,:,2)./L_in).^0.6).*L_out;  
  56. out(:,:,3)=((HDR(:,:,3)./L_in).^0.6).*L_out;  
  57. figure;imshow(out,[]);  
  58. % imwrite(out,'result\AhwahneeGreatLounge_small.png');  
最後實現的LDR影象有一些還是有光暈的,不知是程式的問題,還是論文字身的問題。