Skip to content

Latest commit

 

History

History
28 lines (23 loc) · 884 Bytes

README.md

File metadata and controls

28 lines (23 loc) · 884 Bytes
Raw

NIvsCI

Prerequisites

  • Python 2
  • NVIDIA GPU + CUDA cuDNN
  • PyTorch 0.3.1

Run

  • Normal training: python normal_training.py
  • Construct negative samples in the offline manner: python constructNegativeSamples.py
  • Enhanced training: python enhanced_training.py
  • Test the results: python test.py

Citation

If you find this useful for your research, please use the following.

@INPROCEEDINGS{quan2019enhance, 
  author={W. {Quan} and K. {Wang} and D. {Yan} and D. {Pellerin} and X. {Zhang}}, 
  booktitle={Proceedings of the International Symposium on Image and Signal Processing and Analysis}, 
  title={Improving the Generalization of Colorized Image Detection with Enhanced Training of CNN}, 
  year={2019}, 
  pages={246-252}
}

Acknowledgments

This code borrows from pytorch-CycleGAN-and-pix2pix.