Skip to content
Paper: CVPR2018, Learning Rich Features for Image Manipulation Detection
Branch: master
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
_image
_pdf 8/19 Aug 19, 2018
data trainable version Jul 30, 2018
default/DIY_dataset/default trainable version Jul 30, 2018
lib
.gitignore trainable version Jul 30, 2018
LICENSE Initial commit Jul 6, 2018
Learning Rich Features for Image Manipulation Dection.pdf paper Jul 6, 2018
README.md 8/19 Aug 19, 2018
demo_two_stream.py 8/19 Aug 19, 2018
main_create_training_set.py fix main_create_training_set Oct 10, 2018
train.py trainable version Jul 30, 2018

README.md

Image_manipulation_detection

Paper: CVPR2018, Learning Rich Features for Image Manipulation Detection
Code based on Faster-RCNN

This is a rough implementation of the paper. Since I do not have a titan gpu, I made some modifications on the algorithm, but you can easily change them back if you want the exact setting from the paper.

Environment

Python 3.6 TensorFlow 1.8.0

Setup

  • Download vgg16 pre-trained weights from here
    • save to /data/imagenet_weights/vgg16.ckpt
  • Two-stream neural network model: lib/nets/vgg16.py
    • noise stream's weights are randomly initialized
    • for accurate prediction, please pre-train noise stream's vgg weights on ImageNet and overwrite the trainable setting of noise stream after SRM conv layer
  • Bounding boxes are predicted by both streams.
    • In the paper, RGB stream alone predicts bbox more accurately, so you may wanna change that as well (also defined in vgg16.py)
  • Use main_create_training_set.py to create training set from PASCAL VOC dataset.
    • The generated dataset will follow the pascal voc style, which is also required by train.py
  • Tensorboard file will be save at /default
  • Weights will be save to /default/DIY_detaset/default

Note

The code requires a large memory GPU. If you do not have a 6G+ GPU, please reduce the number of noise stream conv layers for training.

Demo results

Dataset size: 10000, epoch: 3

Finally

I will update this repo a few weeks later after I installed the new GPU

You can’t perform that action at this time.