package train;

import java.awt.Color;

import java.awt.Label;

import java.awt.image.BufferedImage;

import java.io.File;

import java.io.IOException;

import java.nio.file.Files;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import javax.imageio.ImageIO;

import com.jhlabs.image.ScaleFilter;

public class ImageUtil {

/**

* 判断是否位黑色像素

* @param rgb

* @return

*/

public static boolean isBlack(int rgb) {

Color color = new Color(rgb);

if (color.getRed() + color.getGreen() + color.getBlue() <= 300) {

return true;

}

return false;

}

/**

* 缩放图片,默认16x16

* @param img

* @return

*/

public static BufferedImage scaleImage(BufferedImage img) {

return scaleImage(img, 16, 16);

}

/**

* 缩放图片

* @param img

* @param width

* @param height

* @return

*/

public static BufferedImage scaleImage(BufferedImage img, int width, int height){

ScaleFilter sf = new ScaleFilter(width,height);

BufferedImage imgdest = new BufferedImage(width, height, img.getType());

return sf.filter(img, imgdest);

}

/**

* 获得训练集图片的分类，如a-12.jpg，返回a

* @param filename

* @return

*/

public static String getImgClass(String filename){

String[] arr = filename.split("-");

if (arr != null) {

return arr[0];

}else{

return "";

}

}

}