Java 比较两张图片的相似度

1. import java.awt.image.BufferedImage;
2. import java.io.File;
3. import javax.imageio.ImageIO;
4. /**
5. * 比较两张图片的相似度
6. * @author Guihua
7. *
8. */
9. public class BMPLoader {
10. // 改变成二进制码
11. public static String[][] getPX(String args) {
12. int[] rgb = new int[3];
13. File file = new File(args);
14. BufferedImage bi = null;
15. try {
16. bi = ImageIO.read(file);
17. } catch (Exception e) {
18. e.printStackTrace();
19. }
20. int width = bi.getWidth();
21. int height = bi.getHeight();
22. int minx = bi.getMinX();
23. int miny = bi.getMinY();
24. String[][] list = new String[width][height];
25. for (int i = minx; i < width; i++) {
26. for (int j = miny; j < height; j++) {
27. int pixel = bi.getRGB(i, j);
28. rgb[0] = (pixel & 0xff0000) >> 16;
29. rgb[1] = (pixel & 0xff00) >> 8;
30. rgb[2] = (pixel & 0xff);
31. list[i][j] = rgb[0] + "," + rgb[1] + "," + rgb[2];
32. }
33. }
34. return list;
35. }
36. public static void compareImage(String imgPath1, String imgPath2){
37. String[] images = {imgPath1, imgPath2};
38. if (images.length == 0) {
39. System.out.println("Usage >java BMPLoader ImageFile.bmp");
40. System.exit(0);
41. }
42. // 分析图片相似度 begin
43. String[][] list1 = getPX(images[0]);
44. String[][] list2 = getPX(images[1]);
45. int xiangsi = 0;
46. int busi = 0;
47. int i = 0, j = 0;
48. for (String[] strings : list1) {
49. if ((i + 1) == list1.length) {
50. continue;
51. }
52. for (int m=0; m<strings.length; m++) {
53. try {
54. String[] value1 = list1[i][j].toString().split(",");
55. String[] value2 = list2[i][j].toString().split(",");
56. int k = 0;
57. for (int n=0; n<value2.length; n++) {
58. if (Math.abs(Integer.parseInt(value1[k]) - Integer.parseInt(value2[k])) < 5) {
59. xiangsi++;
60. } else {
61. busi++;
62. }
63. }
64. } catch (RuntimeException e) {
65. continue;
66. }
67. j++;
68. }
69. i++;
70. }
71. list1 = getPX(images[1]);
72. list2 = getPX(images[0]);
73. i = 0;
74. j = 0;
75. for (String[] strings : list1) {
76. if ((i + 1) == list1.length) {
77. continue;
78. }
79. for (int m=0; m<strings.length; m++) {
80. try {
81. String[] value1 = list1[i][j].toString().split(",");
82. String[] value2 = list2[i][j].toString().split(",");
83. int k = 0;
84. for (int n=0; n<value2.length; n++) {
85. if (Math.abs(Integer.parseInt(value1[k]) - Integer.parseInt(value2[k])) < 5) {
86. xiangsi++;
87. } else {
88. busi++;
89. }
90. }
91. } catch (RuntimeException e) {
92. continue;
93. }
94. j++;
95. }
96. i++;
97. }
98. String baifen = "";
99. try {
100. baifen = ((Double.parseDouble(xiangsi + "") / Double.parseDouble((busi + xiangsi) + "")) + "");
101. baifen = baifen.substring(baifen.indexOf(".") + 1, baifen.indexOf(".") + 3);
102. } catch (Exception e) {
103. baifen = "0";
104. }
105. if (baifen.length() <= 0) {
106. baifen = "0";
107. }
108. if(busi == 0){
109. baifen="100";
110. }
111. System.out.println("相似像素数量：" + xiangsi + " 不相似像素数量：" + busi + " 相似率：" + Integer.parseInt(baifen) + "%");
112. }
113. public static void main(String[] args){
114. BMPLoader.compareImage("E:\\12.bmp", "E:\\1.bmp");
115. }
116. }