ct_ui.py

#!/usr/bin/python

"""
Hello world of pyqt
"""

import sys
import cv2
from PyQt4.QtGui import *
from PyQt4.QtCore import *
from color_transfer import color_transfer as ct


class ImageView(QLabel):

    def __init__(self, image_path, parent=None):
        super(ImageView, self).__init__(parent)
        self.setGeometry(0, 0, 500, 500)
        pixmap = QPixmap(image_path)
        pixmap = pixmap.scaled(500, 500, 1)
        self.setPixmap(pixmap)

    def setView(self):
        print 'set view'
        self.setGeometry(0, 0, 500, 500)
        pixmap = QPixmap(outpic)
        pixmap = pixmap.scaled(500, 500, 1)
        self.setPixmap(pixmap)



pic1 = '/Users/jiezhi/Documents/python/opencv/color_transfer/images/ocean_sunset.jpg'
# pic1 = '/Users/jiezhi/Documents/python/opencv/color_transfer/images/autumn.jpg'
pic2 = '/Users/jiezhi/Documents/python/opencv/color_transfer/images/fallingwater.jpg'
outpic = '/Users/jiezhi/Documents/python/opencv/color_transfer/images/tmp.jpg'

def main():
    app = QApplication(sys.argv)
    window = QDialog()
    #window = QMainWindow()
    window.setGeometry(0, 0, 1200, 1200)

    layout = QGridLayout()
    
    img1 = ImageView(pic1)
    img2 = ImageView(pic2)
    img3 = ImageView(pic2)
    button = QPushButton('Click')
    
    layout.addWidget(img1, 1, 0)
    layout.addWidget(img2, 1, 1)
    layout.addWidget(img3, 2, 0)
    layout.addWidget(button, 2, 1)

    source = cv2.imread(pic1)
    target = cv2.imread(pic2)
    transfer = ct(source, target)
    cv2.imwrite(outpic, transfer)

    button.clicked.connect(img3.setView)
    # window.connect(button, SIGNAL("released(int)"), img3.setView)

    # layout.show()
    window.setLayout(layout)
    window.show()

    sys.exit(app.exec_())

def setView():
    print 'hello world'
if __name__ == '__main__':
    main()