shangyingao/blockcnn

BlockCNN: A Deep Network for Artifact Removal and Image Compression

BlockCNN

BlockCNN: A Deep Network for Artifact Removal and Image Compression

This repository containing the implementation of BlockCNN which published in CVPR Workshop 2018. The implementation is in Pytorch. Original paper can be found in this link http://openaccess.thecvf.com/content_cvpr_2018_workshops/w50/html/Maleki_BlockCNN_A_Deep_CVPR_2018_paper.html

Requirments:

• Python 3.6
• Pytorch 0.3
• Torchvision
• Opencv
• Tensorflow 1.3
• Matplotlib

Instalation:

git clone https://github.com/DaniMlk/BlockCNN.git
cd BlockCNN
# [Option 1] To replicate the conda environment:
conda env create -f environment.yml
source activate pytorch
# [Option 2] Install everything globaly.

Using

We used Pascal_VOC 2012 to train our network. To run the code firstly you should put your dataset in the root path which mentioned in the main.py

python main.py

Results

In the below image you can see the output of our network, the input image and the original of it. The result of it shows an amazing fact that our network can enhance the quality of the image significantly.

In the below image you can compassion our output with other states of the art compression methods which shows that our network works in a better way.

TODO List:

For now we are working on this project to improve our results. we got promising results and we plan to publish our new algoirthm for ICCV 2019. You will shoke with our results 🔥

Citing

@InProceedings{Maleki_2018_CVPR_Workshops,
author = {Maleki, Danial and Nadalian, Soheila and Mahdi Derakhshani, Mohammad and Amin Sadeghi, Mohammad},
title = {BlockCNN: A Deep Network for Artifact Removal and Image Compression},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR) Workshops},
month = {June},
year = {2018}
}