Java FileUtils.saveImageResultsToPng方法代码示例

import net.semanticmetadata.lire.utils.FileUtils; //导入方法依赖的package包/类
public void testSearchIndexLarge() throws IOException {

        for (int i = 0; i < 10; i++) {
            int queryDocID = (int) (Math.random() * 800);
//            queryDocID = 877 * (i + 1);
            IndexReader reader = DirectoryReader.open(FSDirectory.open(new File("index-large")));
            // select one feature for the large index:
            int featureIndex = 13;
            int count = 0;
            long ms = System.currentTimeMillis();
            ImageSearchHits hits = searchers[featureIndex].search(reader.document(queryDocID), reader);
            for (int j = 0; j < hits.length(); j++) {
                String fileName = hits.doc(j).getValues(
                        DocumentBuilder.FIELD_NAME_IDENTIFIER)[0];
                System.out.println(hits.score(j) + ": \t" + fileName);
            }
//        FileUtils.saveImageResultsToHtml("GeneralTest_testSearchIndexLarge_", hits, reader.document(10).getValues(DocumentBuilder.FIELD_NAME_IDENTIFIER)[0]);
            FileUtils.saveImageResultsToPng("GeneralTest_testSearchIndexLarge_" + i + "_", hits, reader.document(queryDocID).getValues(DocumentBuilder.FIELD_NAME_IDENTIFIER)[0]);
        }
    }
 

import net.semanticmetadata.lire.utils.FileUtils; //导入方法依赖的package包/类
public void testRerankFilters() throws IOException {
    int queryDocID = (int) (Math.random() * 10000);
    IndexReader reader = DirectoryReader.open(FSDirectory.open(new File("index-large")));
    // select one feature for the large index:
    int featureIndex = 4;
    int count = 0;
    long ms = System.currentTimeMillis();
    ImageSearchHits hits = searchers[featureIndex].search(reader.document(queryDocID), reader);
    RerankFilter rerank = new RerankFilter(featureClasses[0], DocumentBuilder.FIELD_NAME_CEDD);
    LsaFilter lsa = new LsaFilter(featureClasses[0], DocumentBuilder.FIELD_NAME_CEDD);
    FileUtils.saveImageResultsToPng("GeneralTest_rerank_0_old", hits, reader.document(queryDocID).getValues(DocumentBuilder.FIELD_NAME_IDENTIFIER)[0]);
    hits = rerank.filter(hits, reader.document(queryDocID));
    FileUtils.saveImageResultsToPng("GeneralTest_rerank_1_new", hits, reader.document(queryDocID).getValues(DocumentBuilder.FIELD_NAME_IDENTIFIER)[0]);
    hits = lsa.filter(hits, reader.document(queryDocID));
    FileUtils.saveImageResultsToPng("GeneralTest_rerank_2_lsa", hits, reader.document(queryDocID).getValues(DocumentBuilder.FIELD_NAME_IDENTIFIER)[0]);
}
 

import net.semanticmetadata.lire.utils.FileUtils; //导入方法依赖的package包/类
/**
 * There was an error that images with the same score but different documents in the index
 * were not included in the result list. Here's the test for that.
 */
public void testDuplicatesInIndex() throws IOException {
    indexFiles("src\\test\\resources\\images", "index-large-new", 0, true);
    indexFiles("src\\test\\resources\\images", "index-large-new", 0, false);
    indexFiles("src\\test\\resources\\images", "index-large-new", 0, false);

    ImageSearcher s = searchers[0];
    IndexReader reader = DirectoryReader.open(FSDirectory.open(new File("index-large-new")));
    Document query = reader.document(0);
    ImageSearchHits hits = s.search(query, reader);
    FileUtils.saveImageResultsToPng("duplicate_", hits, query.getValues(DocumentBuilder.FIELD_NAME_IDENTIFIER)[0]);
}
 

import net.semanticmetadata.lire.utils.FileUtils; //导入方法依赖的package包/类
public void testOutputSearchResults() throws IOException, InstantiationException, IllegalAccessException {
    IndexReader reader = DirectoryReader.open(FSDirectory.open(new File(indexPath)));
    int docNum = 0; // doc to search for.
    // -----------

    String query = reader.document(docNum).getValues("hash")[0];
    CEDD ceddQuery = new CEDD();
    ceddQuery.setByteArrayRepresentation(reader.document(docNum).getField(DocumentBuilder.FIELD_NAME_CEDD).binaryValue().bytes, reader.document(docNum).getField(DocumentBuilder.FIELD_NAME_CEDD).binaryValue().offset, reader.document(docNum).getField(DocumentBuilder.FIELD_NAME_CEDD).binaryValue().length);

    IndexSearcher searcher = new IndexSearcher(reader);
    TopDocs topDocs = searcher.search(createQuery(query), numImagesEval);
    FileUtils.saveImageResultsToPng("result_lsh", topDocs, reader.document(docNum).getValues(DocumentBuilder.FIELD_NAME_IDENTIFIER)[0], reader);
}
 

import net.semanticmetadata.lire.utils.FileUtils; //导入方法依赖的package包/类
public void testSearchInIndexSurf() throws IOException {
    int[] docIDs = new int[]{7886, 1600, 4611, 4833, 4260, 2044, 7658};
    for (int i : docIDs) {
        IndexReader ir = DirectoryReader.open(FSDirectory.open(indexPath));
        SurfFeatureHistogramBuilder sfh = new SurfFeatureHistogramBuilder(ir);
        VisualWordsImageSearcher vis = new VisualWordsImageSearcher(10, DocumentBuilder.FIELD_NAME_SURF_VISUAL_WORDS);
        Document doc = sfh.getVisualWords(surfBuilder.createDocument(ImageIO.read(new File(queryImage)), queryImage));
        ImageSearchHits hits = vis.search(ir.document(i), ir);
        FileUtils.saveImageResultsToPng("results_bow_no_tf_" + i, hits, queryImage);
    }
}
 

import net.semanticmetadata.lire.utils.FileUtils; //导入方法依赖的package包/类
public void testSearchExternalImageSurf() throws IOException {
    IndexReader ir = DirectoryReader.open(FSDirectory.open(indexPath));
    SurfFeatureHistogramBuilder sfh = new SurfFeatureHistogramBuilder(ir);
    VisualWordsImageSearcher vis = new VisualWordsImageSearcher(10, DocumentBuilder.FIELD_NAME_SURF_VISUAL_WORDS);
    Document doc = sfh.getVisualWords(surfBuilder.createDocument(ImageIO.read(new File(queryImage)), queryImage));
    ImageSearchHits hits = vis.search(doc, ir);
    FileUtils.saveImageResultsToPng("results_bow_surf", hits, queryImage);
}
 

import net.semanticmetadata.lire.utils.FileUtils; //导入方法依赖的package包/类
public void testSearchInIndexSift() throws IOException {
    int[] docIDs = new int[]{0, 10, 23, 35, 56, 77};
    for (int i : docIDs) {
        System.out.println("i = " + i);
        IndexReader ir = DirectoryReader.open(FSDirectory.open(indexPath));
        VisualWordsImageSearcher vis = new VisualWordsImageSearcher(10, DocumentBuilder.FIELD_NAME_SIFT_VISUAL_WORDS);
        ImageSearchHits hits = vis.search(ir.document(i), ir);
        FileUtils.saveImageResultsToPng("results_bow_no_tf_sift_" + i, hits, queryImage);
    }
}
 

import net.semanticmetadata.lire.utils.FileUtils; //导入方法依赖的package包/类
public void testSearchExternalImageSift() throws IOException {
    IndexReader ir = DirectoryReader.open(FSDirectory.open(indexPath));
    SurfFeatureHistogramBuilder sfh = new SurfFeatureHistogramBuilder(ir);
    VisualWordsImageSearcher vis = new VisualWordsImageSearcher(10, DocumentBuilder.FIELD_NAME_SIFT_VISUAL_WORDS);
    Document doc = sfh.getVisualWords(siftBuilder.createDocument(ImageIO.read(new File(queryImage)), queryImage));
    ImageSearchHits hits = vis.search(doc, ir);
    FileUtils.saveImageResultsToPng("results_bow_sift", hits, queryImage);
}
 

import net.semanticmetadata.lire.utils.FileUtils; //导入方法依赖的package包/类
public void testWikiSearchIndex() throws IOException {
    String indexPath = "./bovw-test";
    VisualWordsImageSearcher searcher = new VisualWordsImageSearcher(10,
            DocumentBuilder.FIELD_NAME_SURF_VISUAL_WORDS);
    IndexReader reader = DirectoryReader.open(FSDirectory.open(new File(indexPath)));
    // let's take the first document for a query:
    Document query = reader.document(2);
    ImageSearchHits hits = searcher.search(query, reader);
    // show or analyze your results ....
    FileUtils.saveImageResultsToPng("bovw", hits, query.getValues(DocumentBuilder.FIELD_NAME_IDENTIFIER)[0]);
}