Remove JPEG noise via CNN on Keras and Tensorflow 2.0

Introduction

Simple way to remove JPEG noise via CNN.

Results on JPEG Quality=5 (1-100)

JPEG	Removed	Real

Data

• Download training data from DIV2K
• Get trained model from Google Drive

Environment

python==3.6
tensorflow==2.0

How to use

Train

python train.py

Predict

python predict.py --image test.jpg --model weights.h5