版权声明:作者:上善若水rs 本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。 https://blog.csdn.net/sofeware333/article/details/89571645
private ImageInfo loadTestImage(String path) throws Exception{
File img = new File(path);
if (!img.exists()) {
return null;
}
BufferedImage image = null;
try {
image = ImageIO.read(img);
} catch (Exception e) {
if(image == null){
CMYK.readImage(path);//将CMYK模式转为rgb模式
img = new File(path);
image = ImageIO.read(img);
}
if(image == null){
logger.debug("转换图片失败,源图片路径:"+path+ ";原因:该图片无法转换成rgb");
System.out.println("转换图片失败,源图片路径:"+path+ ";原因:该图片无法转换成rgb");
return null;//如果还是无法转化,则跳过
}
}
if (image == null) {
return null;
}
final int width = image.getWidth();
final int height = image.getHeight();
int[] pix = new int[width * height];
PixelGrabber pg = new PixelGrabber(image, 0, 0, width, height, pix, 0, width);
try {
if (!pg.grabPixels()) {
return null;
}
} catch (InterruptedException e) {
logger.debug("转换图片失败,源图片路径:"+path+ ";原因:该图片无法转换成rgb.");
System.out.println("转换图片失败,源图片路径:"+path+ ";原因:该图片无法转换成rgb.");
}
ImageInfo imageInfo = new ImageInfo();
imageInfo.setWidth(width);
imageInfo.setHeight(height);
byte[] rgb24 = new byte[width * height * 3];
for (int i = 0; i < height; i++) {
for (int j = 0; j < width; j++) {
int idx = width * i + j;
int color = pix[idx]; //获取像素
int red = ((color & 0x00FF0000) >> 16);
int green = ((color & 0x0000FF00) >> 8);
int blue = color & 0x000000FF;
int rgbIdx = idx * 3;
rgb24[rgbIdx] = (byte) red;
rgb24[rgbIdx + 1] = (byte) green;
rgb24[rgbIdx + 2] = (byte) blue;
}
}
imageInfo.setImage(rgb24);
return imageInfo;
}
private static class ImageInfo {
private byte[] image;
private int width;
private int height;
private float score;
public float getScore() {
return score;
}
public void setScore(float score) {
this.score = score;
}
public byte[] getImage() {
return image;
}
public void setImage(byte[] image) {
this.image = image;
}
public int getWidth() {
return width;
}
public void setWidth(int width) {
this.width = width;
}
public int getHeight() {
return height;
}
public void setHeight(int height) {
this.height = height;
}
}