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
- Python 3.6
- Pytorch 0.3
- Torchvision
- Opencv
- Tensorflow 1.3
- Matplotlib
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.
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
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.
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 🔥
@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}
}