CopyMove.CloneDetect Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of image-forensics Show documentation
A library of image forensics algorithms for tampering localization.
The newest version!
package CopyMove;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

/**
 * Copy-move detection using the PatchMatch Algorithm
 *
 * | PatchMatch : A Randomized Correspondence Algorithm for Structural Image Editing
 * | by Connelly Barnes and Eli Shechtman and Adam Finkelstein and Dan B Goldman
 * | ACM Transactions on Graphics (Proc. SIGGRAPH), vol.28, aug-2009
 *
 * This is simply a modified version of the PatchMatch inpainting code by Xavier Philippeau
 * https://github.com/davidchatting/PatchMatch
 *
 *
 * Created by marzampoglou on 12/2/15.
 */

public class CloneDetect {
    //minimum allowed distance between matching patches is (image diagonal)/minDist
    int minDist=15;
    //number of loops per pyramid level for optimization
    int LoopMultiplier = 1;
    //number of Pyramid levels
    int pyramidLevels=3;

    //initial image
    ImageMap initial;

    // Nearest-Neighbor Field
    NNF_NoMask nnf_TargetToSource;

    // patch radius
    int radius;

    // Pyramid of downsampled initial images
    List pyramid;


    public BufferedImage FindClones(BufferedImage input, int blockSize){
        ArrayList NNFList = new ArrayList<>();

        // initial image
        this.initial = new ImageMap(input);

        // patch blockSize
        this.radius = blockSize;

        // working copies
        ImageMap source = initial;

        System.out.println("build pyramid of images...");

        // build pyramid of downscaled images
        this.pyramid = new ArrayList();
        this.pyramid.add(source);
        //while(source.W>blockSize && source.H>blockSize) {
        for (int ii=0;ii=0;level--) {
            System.out.println("\n*** Processing -  Zoom 1:" + (1 << level) + " ***");
            // create Nearest-Neighbor Fields (direct and reverse)
            source = this.pyramid.get(level);
            target = source;

            System.out.println("initialize NNF...");
            if (level == maxlevel - 1) {
                // at first,  use random data as initial guess
                nnf_TargetToSource = new NNF_NoMask(target, source, blockSize);
                nnf_TargetToSource.randomizeDist(minDist);
                NNF_NoMask new_nnf = new NNF_NoMask(target, source, blockSize);
                new_nnf.initialize(nnf_TargetToSource);
                new_nnf.minimizeDistance(LoopMultiplier *(level+ 1),minDist);
                nnf_TargetToSource=new_nnf;
            } else {
                // then, we use the rebuilt (upscaled) target
                // and reuse the previous NNF as initial guess
                //nnf_TargetToSource.randomize();
                NNF_NoMask new_nnf = new NNF_NoMask(target, source, blockSize);
                new_nnf.initialize(nnf_TargetToSource);
                new_nnf.minimizeDistance(LoopMultiplier *(level+ 1),minDist);
                nnf_TargetToSource = new_nnf;
            }
        }

        String output = "";

        try {
            File file = new File("/home/marzampoglou/Desktop/X.txt");

            if (!file.exists()) {
                file.createNewFile();
            }
            FileWriter fw = new FileWriter(file.getAbsoluteFile());
            BufferedWriter bw = new BufferedWriter(fw);
            for (int ii = 0; ii < nnf_TargetToSource.getField().length; ii++) {
                for (int jj = 0; jj < nnf_TargetToSource.getField()[0].length; jj++) {
                    output=output+nnf_TargetToSource.getField()[ii][jj][0] + " ";
                }
                output=output+"\n";
            }
            bw.write(output);
            bw.close();


            output="";
            file = new File("/home/marzampoglou/Desktop/Y.txt");
            fw = new FileWriter(file.getAbsoluteFile());
            bw = new BufferedWriter(fw);
            if (!file.exists()) {
                file.createNewFile();
            }

            for (int ii = 0; ii < nnf_TargetToSource.getField().length; ii++) {
                for (int jj = 0; jj < nnf_TargetToSource.getField()[0].length; jj++) {
                    output=output+nnf_TargetToSource.getField()[ii][jj][1] + " ";
                }
                output=output+"\n";
            }
            bw.write(output);
            bw.close();


            output="";
            file = new File("/home/marzampoglou/Desktop/Err.txt");
            fw = new FileWriter(file.getAbsoluteFile());
            bw = new BufferedWriter(fw);
            for (int ii = 0; ii < nnf_TargetToSource.getField().length; ii++) {
                for (int jj = 0; jj < nnf_TargetToSource.getField()[0].length; jj++) {
                    output=output+nnf_TargetToSource.getField()[ii][jj][2] + " ";
                }
                output=output+"\n";
            }
            bw.write(output);
            bw.close();

        }
        catch (Exception e){
            e.printStackTrace();
        }
        /*
        */

        BufferedImage out= DrawGrey(nnf_TargetToSource.linearFilterField());
        nnf_TargetToSource.removeLowVar(7,2.5);

        //display(output);
        try {
            ImageIO.write(out, "PNG", new File("/home/marzampoglou/Pictures/Reveal/ManipulationOutput/linearfilt.png"));
        } catch (IOException e) {
            e.printStackTrace();
        }

        return DrawGrey2(nnf_TargetToSource.getField());
    }

    private BufferedImage DrawGrey(int[][] MapIn){
        BufferedImage ImOut = new BufferedImage(MapIn.length,MapIn[0].length, BufferedImage.TYPE_INT_ARGB) ;
        for (int ii=0;ii255) tmpVal=255;
                int value = 0xFF000000 | tmpVal << 16 | tmpVal << 8 | tmpVal;
                ImOut.setRGB(ii,jj,value);
                //System.out.print(tmpVal + " ");
            }
            //System.out.println();
        }
        return ImOut;
    }

    private BufferedImage DrawGrey2(int[][][] MapIn){
        BufferedImage ImOut = new BufferedImage(MapIn.length,MapIn[0].length, BufferedImage.TYPE_INT_ARGB) ;
        for (int ii=0;ii255) tmpVal=255;
                int value = 0xFF000000 | tmpVal << 16 | tmpVal << 8 | tmpVal;
                ImOut.setRGB(ii,jj,value);
                //System.out.print(tmpVal + " ");
            }
            //System.out.println();
        }
        return ImOut;
    }


    }