A Python implementation of RAISR. Supporting color images.
You can install most of the following packages using pip.
Put your training images in the train
directory. The training images are the high resolution (HR) ones. Run the following command to start training.
python train.py
In the training stage, the program virtually downscales the high resolution images. The program then trains the model using the downscaled version images and the original HR images. The learned filters will be saved in the root directory of the project.
Put your testing images in the test
directory. Basically, you can use some low resolution (LR) images as your testing images. By running the following command, the HR version of your testing images will be generated using RAISR.
python test.py
The result will be saved in the results
directory.
Uncomment the following line in train.py
.
# filterplot(h, R, Qangle, Qstrength, Qcoherence, patchsize)
Uncomment the following line in test.py
.
# plt.show()
Comparing between original image, bilinear interpolation and RAISR:
Origin | Bilinear Interpolation | RAISR |
---|---|---|
![]() |
![]() |
![]() |
Other results using images taken from BSDS500 database and ArTe-Lab 1D Medium Barcode Dataset:
Origin | RAISR |
---|---|
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
- Y. Romano, J. Isidoro and P. Milanfar, "RAISR: Rapid and Accurate Image Super Resolution" in IEEE Transactions on Computational Imaging, vol. 3, no. 1, pp. 110-125, March 2017.
- P. Arbelaez, M. Maire, C. Fowlkes and J. Malik, "Contour Detection and Hierarchical Image Segmentation", IEEE TPAMI, Vol. 33, No. 5, pp. 898-916, May 2011.
- Alessandro Zamberletti, Ignazio Gallo and Simone Albertini, "Robust Angle Invariant 1D Barcode Detection", Proceedings of the 2nd Asian Conference on Pattern Recognition (ACPR), Okinawa, Japan, 2013
MIT. Copyright (c) 2017 James Chen