Java图片上查找图片算法
2023-09-11 14:20:53 时间
之前用按键精灵写过一些游戏辅助，里面有个函数叫FindPic，就上在屏幕范围查找给定的一张图片，返回查找到的坐标位置。
现在，Java来实现这个函数类似的功能。
算法描述：
屏幕截图，得到图A，(查找的目标图片为图B)；
遍历图A的像素点，根据图B的尺寸，得到图B四个角映射到图A上的四个点；
得到的四个点与图B的四个角像素点的值比较。如果四个点一样，执行步骤4；否则，回到步骤2继续；
进一步对比，将映射范围内的全部点与图B全部的点比较。如果全部一样，则说明图片已找到；否则，回到步骤2继续；
这里，像素之间的比较是通过BufferedImage对象获取每个像素的RGB值来比较的。如下，将BufferedImage转换为int二维数组：
6 public static int[][] getImageGRB(BufferedImage bfImage) { 7 int width = bfImage.getWidth(); 8 int height = bfImage.getHeight(); 9 int[][] result = new int[height][width]; 10 for (int h = 0; h height; h++) { 11 for (int w = 0; w width; w++) { 12 //使用getRGB(w, h)获取该点的颜色值是ARGB，而在实际应用中使用的是RGB，所以需要将ARGB转化成RGB，即bufImg.getRGB(w, h) 0xFFFFFF。 13 result[h][w] = bfImage.getRGB(w, h) 0xFFFFFF; 14 } 15 } 16 return result; 17 }
比较两个像素点的RGB值是否相同，是通过异或操作比较的（据说比==效率更高），如果异或操作后得到的值为0，说明两个像素点的RGB一样，否则不一样。
下面附上算法完整java代码：
20 BufferedImage screenShotImage; //屏幕截图 21 BufferedImage keyImage; //查找目标图片 23 int scrShotImgWidth; //屏幕截图宽度 24 int scrShotImgHeight; //屏幕截图高度 26 int keyImgWidth; //查找目标图片宽度 27 int keyImgHeight; //查找目标图片高度 29 int[][] screenShotImageRGBData; //屏幕截图RGB数据 30 int[][] keyImageRGBData; //查找目标图片RGB数据 32 int[][][] findImgData; //查找结果，目标图标位于屏幕截图上的坐标数据 35 public ImageFindDemo(String keyImagePath) { 36 screenShotImage = this.getFullScreenShot(); 37 keyImage = this.getBfImageFromPath(keyImagePath); 38 screenShotImageRGBData = this.getImageGRB(screenShotImage); 39 keyImageRGBData = this.getImageGRB(keyImage); 40 scrShotImgWidth = screenShotImage.getWidth(); 41 scrShotImgHeight = screenShotImage.getHeight(); 42 keyImgWidth = keyImage.getWidth(); 43 keyImgHeight = keyImage.getHeight(); 45 //开始查找 46 this.findImage(); 48 } 50 /** 51 * 全屏截图 52 * @return 返回BufferedImage 53 */ 54 public BufferedImage getFullScreenShot() { 55 BufferedImage bfImage = null; 56 int width = (int) Toolkit.getDefaultToolkit().getScreenSize().getWidth(); 57 int height = (int) Toolkit.getDefaultToolkit().getScreenSize().getHeight(); 58 try { 59 Robot robot = new Robot(); 60 bfImage = robot.createScreenCapture(new Rectangle(0, 0, width, height)); 61 } catch (AWTException e) { 62 e.printStackTrace(); 63 } 64 return bfImage; 65 } 67 /** 68 * 从本地文件读取目标图片 69 * @param keyImagePath - 图片绝对路径 70 * @return 本地图片的BufferedImage对象 71 */ 72 public BufferedImage getBfImageFromPath(String keyImagePath) { 73 BufferedImage bfImage = null; 74 try { 75 bfImage = ImageIO.read(new File(keyImagePath)); 76 } catch (IOException e) { 77 e.printStackTrace(); 78 } 79 return bfImage; 80 } 82 /** 83 * 根据BufferedImage获取图片RGB数组 84 * @param bfImage 85 * @return 86 */ 87 public int[][] getImageGRB(BufferedImage bfImage) { 88 int width = bfImage.getWidth(); 89 int height = bfImage.getHeight(); 90 int[][] result = new int[height][width]; 91 for (int h = 0; h height; h++) { 92 for (int w = 0; w width; w++) { 93 //使用getRGB(w, h)获取该点的颜色值是ARGB，而在实际应用中使用的是RGB，所以需要将ARGB转化成RGB，即bufImg.getRGB(w, h) 0xFFFFFF。 94 result[h][w] = bfImage.getRGB(w, h) 0xFFFFFF; 95 } 96 } 97 return result; 98 } 101 /** 102 * 查找图片 103 */ 104 public void findImage() { 105 findImgData = new int[keyImgHeight][keyImgWidth][2]; 106 //遍历屏幕截图像素点数据 107 for(int y=0; y scrShotImgHeight-keyImgHeight; y++) { 108 for(int x=0; x scrShotImgWidth-keyImgWidth; x++) { 109 //根据目标图的尺寸，得到目标图四个角映射到屏幕截图上的四个点， 110 //判断截图上对应的四个点与图B的四个角像素点的值是否相同， 111 //如果相同就将屏幕截图上映射范围内的所有的点与目标图的所有的点进行比较。 112 if((keyImageRGBData[0][0]^screenShotImageRGBData[y][x])==0 113 (keyImageRGBData[0][keyImgWidth-1]^screenShotImageRGBData[y][x+keyImgWidth-1])==0 114 (keyImageRGBData[keyImgHeight-1][keyImgWidth-1]^screenShotImageRGBData[y+keyImgHeight-1][x+keyImgWidth-1])==0 115 (keyImageRGBData[keyImgHeight-1][0]^screenShotImageRGBData[y+keyImgHeight-1][x])==0) { 117 boolean isFinded = isMatchAll(y, x); 118 //如果比较结果完全相同，则说明图片找到，填充查找到的位置坐标数据到查找结果数组。 119 if(isFinded) { 120 for(int h=0; h keyImgHeight; h++) { 121 for(int w=0; w keyImgWidth; w++) { 122 findImgData[h][w][0] = y+h; 123 findImgData[h][w][1] = x+w; 124 } 125 } 126 return; 127 } 128 } 129 } 130 } 131 } 133 /** 134 * 判断屏幕截图上目标图映射范围内的全部点是否全部和小图的点一一对应。 135 * @param y - 与目标图左上角像素点想匹配的屏幕截图y坐标 136 * @param x - 与目标图左上角像素点想匹配的屏幕截图x坐标 137 * @return 138 */ 139 public boolean isMatchAll(int y, int x) { 140 int biggerY = 0; 141 int biggerX = 0; 142 int xor = 0; 143 for(int smallerY=0; smallerY keyImgHeight; smallerY++) { 144 biggerY = y+smallerY; 145 for(int smallerX=0; smallerX keyImgWidth; smallerX++) { 146 biggerX = x+smallerX; 147 if(biggerY =scrShotImgHeight || biggerX =scrShotImgWidth) { 148 return false; 149 } 150 xor = keyImageRGBData[smallerY][smallerX]^screenShotImageRGBData[biggerY][biggerX]; 151 if(xor!=0) { 152 return false; 153 } 154 } 155 biggerX = x; 156 } 157 return true; 158 } 160 /** 161 * 输出查找到的坐标数据 162 */ 163 private void printFindData() { 164 for(int y=0; y keyImgHeight; y++) { 165 for(int x=0; x keyImgWidth; x++) { 166 System.out.print("("+this.findImgData[y][x][0]+", "+this.findImgData[y][x][1]+")"); 167 } 168 System.out.println(); 169 } 170 } 173 public static void main(String[] args) { 174 String keyImagePath = "D:/key.png"; 175 ImageFindDemo demo = new ImageFindDemo(keyImagePath); 176 demo.printFindData(); 177 } 179 }
这种算法是精确比较，只要有一个像素点有差异，就会找不到图片。当然，如果想指定一个比较的精确度，我也有个思路，就是在算法步骤4比较映射范围内全部像素点的时候做个统计，如果90%的点都相同，那就是说精确度是0.9。 另外，可能还要考虑效率问题，不过，我在我的应用场景中并不太在意效率。如果有朋友看到这篇文章，对这个话题有更好的想法，请留言。 最新内容请见作者的GitHub页：http://qaseven.github.io/
2023年阿里高频Java面试题：分布式+中间件+高并发+算法+数据库 又到了一年一度的金九银十，互联网行业竞争是一年比一年严峻，作为工程师的我们唯有不停地学习，不断的提升自己才能保证自己的核心竞争力从而拿到更好的薪水，进入心仪的企业（阿里、字节、美团、腾讯.....）
6 public static int[][] getImageGRB(BufferedImage bfImage) { 7 int width = bfImage.getWidth(); 8 int height = bfImage.getHeight(); 9 int[][] result = new int[height][width]; 10 for (int h = 0; h height; h++) { 11 for (int w = 0; w width; w++) { 12 //使用getRGB(w, h)获取该点的颜色值是ARGB，而在实际应用中使用的是RGB，所以需要将ARGB转化成RGB，即bufImg.getRGB(w, h) 0xFFFFFF。 13 result[h][w] = bfImage.getRGB(w, h) 0xFFFFFF; 14 } 15 } 16 return result; 17 }
比较两个像素点的RGB值是否相同，是通过异或操作比较的（据说比==效率更高），如果异或操作后得到的值为0，说明两个像素点的RGB一样，否则不一样。
下面附上算法完整java代码：
20 BufferedImage screenShotImage; //屏幕截图 21 BufferedImage keyImage; //查找目标图片 23 int scrShotImgWidth; //屏幕截图宽度 24 int scrShotImgHeight; //屏幕截图高度 26 int keyImgWidth; //查找目标图片宽度 27 int keyImgHeight; //查找目标图片高度 29 int[][] screenShotImageRGBData; //屏幕截图RGB数据 30 int[][] keyImageRGBData; //查找目标图片RGB数据 32 int[][][] findImgData; //查找结果，目标图标位于屏幕截图上的坐标数据 35 public ImageFindDemo(String keyImagePath) { 36 screenShotImage = this.getFullScreenShot(); 37 keyImage = this.getBfImageFromPath(keyImagePath); 38 screenShotImageRGBData = this.getImageGRB(screenShotImage); 39 keyImageRGBData = this.getImageGRB(keyImage); 40 scrShotImgWidth = screenShotImage.getWidth(); 41 scrShotImgHeight = screenShotImage.getHeight(); 42 keyImgWidth = keyImage.getWidth(); 43 keyImgHeight = keyImage.getHeight(); 45 //开始查找 46 this.findImage(); 48 } 50 /** 51 * 全屏截图 52 * @return 返回BufferedImage 53 */ 54 public BufferedImage getFullScreenShot() { 55 BufferedImage bfImage = null; 56 int width = (int) Toolkit.getDefaultToolkit().getScreenSize().getWidth(); 57 int height = (int) Toolkit.getDefaultToolkit().getScreenSize().getHeight(); 58 try { 59 Robot robot = new Robot(); 60 bfImage = robot.createScreenCapture(new Rectangle(0, 0, width, height)); 61 } catch (AWTException e) { 62 e.printStackTrace(); 63 } 64 return bfImage; 65 } 67 /** 68 * 从本地文件读取目标图片 69 * @param keyImagePath - 图片绝对路径 70 * @return 本地图片的BufferedImage对象 71 */ 72 public BufferedImage getBfImageFromPath(String keyImagePath) { 73 BufferedImage bfImage = null; 74 try { 75 bfImage = ImageIO.read(new File(keyImagePath)); 76 } catch (IOException e) { 77 e.printStackTrace(); 78 } 79 return bfImage; 80 } 82 /** 83 * 根据BufferedImage获取图片RGB数组 84 * @param bfImage 85 * @return 86 */ 87 public int[][] getImageGRB(BufferedImage bfImage) { 88 int width = bfImage.getWidth(); 89 int height = bfImage.getHeight(); 90 int[][] result = new int[height][width]; 91 for (int h = 0; h height; h++) { 92 for (int w = 0; w width; w++) { 93 //使用getRGB(w, h)获取该点的颜色值是ARGB，而在实际应用中使用的是RGB，所以需要将ARGB转化成RGB，即bufImg.getRGB(w, h) 0xFFFFFF。 94 result[h][w] = bfImage.getRGB(w, h) 0xFFFFFF; 95 } 96 } 97 return result; 98 } 101 /** 102 * 查找图片 103 */ 104 public void findImage() { 105 findImgData = new int[keyImgHeight][keyImgWidth][2]; 106 //遍历屏幕截图像素点数据 107 for(int y=0; y scrShotImgHeight-keyImgHeight; y++) { 108 for(int x=0; x scrShotImgWidth-keyImgWidth; x++) { 109 //根据目标图的尺寸，得到目标图四个角映射到屏幕截图上的四个点， 110 //判断截图上对应的四个点与图B的四个角像素点的值是否相同， 111 //如果相同就将屏幕截图上映射范围内的所有的点与目标图的所有的点进行比较。 112 if((keyImageRGBData[0][0]^screenShotImageRGBData[y][x])==0 113 (keyImageRGBData[0][keyImgWidth-1]^screenShotImageRGBData[y][x+keyImgWidth-1])==0 114 (keyImageRGBData[keyImgHeight-1][keyImgWidth-1]^screenShotImageRGBData[y+keyImgHeight-1][x+keyImgWidth-1])==0 115 (keyImageRGBData[keyImgHeight-1][0]^screenShotImageRGBData[y+keyImgHeight-1][x])==0) { 117 boolean isFinded = isMatchAll(y, x); 118 //如果比较结果完全相同，则说明图片找到，填充查找到的位置坐标数据到查找结果数组。 119 if(isFinded) { 120 for(int h=0; h keyImgHeight; h++) { 121 for(int w=0; w keyImgWidth; w++) { 122 findImgData[h][w][0] = y+h; 123 findImgData[h][w][1] = x+w; 124 } 125 } 126 return; 127 } 128 } 129 } 130 } 131 } 133 /** 134 * 判断屏幕截图上目标图映射范围内的全部点是否全部和小图的点一一对应。 135 * @param y - 与目标图左上角像素点想匹配的屏幕截图y坐标 136 * @param x - 与目标图左上角像素点想匹配的屏幕截图x坐标 137 * @return 138 */ 139 public boolean isMatchAll(int y, int x) { 140 int biggerY = 0; 141 int biggerX = 0; 142 int xor = 0; 143 for(int smallerY=0; smallerY keyImgHeight; smallerY++) { 144 biggerY = y+smallerY; 145 for(int smallerX=0; smallerX keyImgWidth; smallerX++) { 146 biggerX = x+smallerX; 147 if(biggerY =scrShotImgHeight || biggerX =scrShotImgWidth) { 148 return false; 149 } 150 xor = keyImageRGBData[smallerY][smallerX]^screenShotImageRGBData[biggerY][biggerX]; 151 if(xor!=0) { 152 return false; 153 } 154 } 155 biggerX = x; 156 } 157 return true; 158 } 160 /** 161 * 输出查找到的坐标数据 162 */ 163 private void printFindData() { 164 for(int y=0; y keyImgHeight; y++) { 165 for(int x=0; x keyImgWidth; x++) { 166 System.out.print("("+this.findImgData[y][x][0]+", "+this.findImgData[y][x][1]+")"); 167 } 168 System.out.println(); 169 } 170 } 173 public static void main(String[] args) { 174 String keyImagePath = "D:/key.png"; 175 ImageFindDemo demo = new ImageFindDemo(keyImagePath); 176 demo.printFindData(); 177 } 179 }
这种算法是精确比较，只要有一个像素点有差异，就会找不到图片。当然，如果想指定一个比较的精确度，我也有个思路，就是在算法步骤4比较映射范围内全部像素点的时候做个统计，如果90%的点都相同，那就是说精确度是0.9。 另外，可能还要考虑效率问题，不过，我在我的应用场景中并不太在意效率。如果有朋友看到这篇文章，对这个话题有更好的想法，请留言。 最新内容请见作者的GitHub页：http://qaseven.github.io/
2023年阿里高频Java面试题：分布式+中间件+高并发+算法+数据库 又到了一年一度的金九银十，互联网行业竞争是一年比一年严峻，作为工程师的我们唯有不停地学习，不断的提升自己才能保证自己的核心竞争力从而拿到更好的薪水，进入心仪的企业（阿里、字节、美团、腾讯.....）
相关文章
- Java实现蓝桥杯VIP算法训练 奇变的字符串
- Java实现 LeetCode 12 整数转罗马数字
- Java实现 蓝桥杯VIP 算法提高 打水问题
- Java实现 蓝桥杯VIP 算法训练 数列
- Java实现 蓝桥杯VIP 算法训练 成绩的等级输出
- Java实现 蓝桥杯VIP 算法训练 猴子分苹果
- Java实现 蓝桥杯 算法提高 合并石子
- Java实现 蓝桥杯 算法训练 最大的算式
- Java实现 蓝桥杯 算法训练 删除数组零元素
- Java实现 蓝桥杯 算法训练 Torry的困惑（基本型）
- （Java实现） 活动选择
- Java 蓝桥杯 算法训练 貌似化学
- Java 蓝桥杯 算法训练 字符串的展开 （JAVA语言实现)
- Java 蓝桥杯 算法训练 字符串的展开 （JAVA语言实现)
- java 多线程下载
- Java中Properties类的操作
- java复习-多线程
- Java设计模式之建造者Builder模式代码示例
- Java工具类--jwt微服务token签发与验证
- Atitit java opencv 捕获视频
- Atitit.java图片图像处理attilax总结 BufferedImage extends java.awt.Image获取图像像素点image.getRGB(i, lineIndex); 图片剪辑/AtiPlatf_cms/src/com/attilax/img/imgx.javacutImage图片处理titit 判断判断一张图片是否包含另一张小图片 atitit 图片去噪算法的原理与
- thrift和java交互案例和结果
- Java 算法合并 Geoserver 切片生成指北针图片：高效、优雅解决地图数据可视化问题
- java的重写规则