基于Lire库搜索相似图片源码

package com.nii.sim; import net.semanticmetadata.lire.builders.DocumentBuilder; import net.semanticmetadata.lire.builders.GlobalDocumentBuilder; import net.semanticmetadata.lire.imageanalysis.features.global.CEDD; import net.semanticmetadata.lire.searchers.GenericFastImageSearcher; import net.semanticmetadata.lire.searchers.ImageSearchHits; import net.semanticmetadata.lire.searchers.ImageSearcher; import net.semanticmetadata.lire.utils.FileUtils; import org.apache.lucene.analysis.core.WhitespaceAnalyzer; import org.apache.lucene.document.Document; import org.apache.lucene.index.DirectoryReader; import org.apache.lucene.index.IndexReader; import org.apache.lucene.index.IndexWriter; import org.apache.lucene.index.IndexWriterConfig; import org.apache.lucene.store.FSDirectory; import javax.imageio.ImageIO; import java.awt.image.BufferedImage; import java.io.FileInputStream; import java.io.IOException; import java.nio.file.Paths; import java.util.Iterator; import java.util.List; /** * @author wzj * @create 2018-07-22 11:16 **/ public class ImageSimilarityTest { /** * 图片保存的路径 */ private static final String IMAGE_PATH = "H:\\JAVA\\ImageSim\\conf\\image"; /** * 索引保存目录 */ private static final String INDEX_PATH = "H:\\JAVA\\ImageSim\\conf\\index"; public static void main(String[] args) throws IOException { //indexImages(); searchSimilarityImage(); } private static void indexImages() throws IOException { List<String> images = FileUtils.getAllImages(Paths.get(IMAGE_PATH).toFile(), true); GlobalDocumentBuilder globalDocumentBuilder = new GlobalDocumentBuilder(false, false); globalDocumentBuilder.addExtractor(CEDD.class); IndexWriterConfig conf = new IndexWriterConfig(new WhitespaceAnalyzer()); IndexWriter indexWriter = new IndexWriter(FSDirectory.open(Paths.get(INDEX_PATH)), conf); for (Iterator<String> it = images.iterator(); it.hasNext(); ) { String imageFilePath = it.next(); System.out.println("Indexing " + imageFilePath); BufferedImage img = ImageIO.read(new FileInputStream(imageFilePath)); Document document = globalDocumentBuilder.createDocument(img, imageFilePath); indexWriter.addDocument(document); } indexWriter.close(); System.out.println("Create index image successful."); } private static void searchSimilarityImage() throws IOException { IndexReader ir = DirectoryReader.open(FSDirectory.open(Paths.get(INDEX_PATH))); ImageSearcher searcher = new GenericFastImageSearcher(5, CEDD.class); String inputImagePath = "H:\\JAVA\\ImageSim\\conf\\image\\5.png"; BufferedImage img = ImageIO.read(Paths.get(inputImagePath).toFile()); ImageSearchHits hits = searcher.search(img, ir); for (int i = 0; i < hits.length(); i++) { String fileName = ir.document(hits.documentID(i)).getValues(DocumentBuilder.FIELD_NAME_IDENTIFIER)[0]; System.out.println(hits.score(i) + ": \t" + fileName); } } }