import sys
|
|
from PySide6.QtGui import QPixmap, QImage
|
#from PIL.ImageQt import QPixmap, QImage
|
|
from PySide6.QtWidgets import QApplication, QMainWindow, QFileDialog
|
import numpy as np
|
from PIL import Image
|
|
from imgdemo.ui_mainwindow import Ui_MainWindow
|
|
class MainWindow(QMainWindow):
|
def __init__(self):
|
super(MainWindow, self).__init__()
|
self.saved_file = ('', '')
|
self.chosen_file = ('', '')
|
self.ui = Ui_MainWindow()
|
self.ui.setupUi(self)
|
self.ui.actionChoose_Image.triggered.connect(self.choose_image)
|
self.ui.actionSave_Image.triggered.connect(self.save_image)
|
|
def choose_image(self):
|
self.chosen_file = ('', '') # reset (for safe)
|
self.chosen_file = QFileDialog.getOpenFileName(None, "Choose Image", "",
|
"Images (*.png *.xpm *.jpg *.tiff) ;; All Filetypes (*.*)")
|
print(self.chosen_file)
|
if self.chosen_file[0] != '': # user has chosen a file
|
self.display_image()
|
|
|
def save_image(self):
|
self.saved_file = ('', '') # reset for safe
|
self.saved_file = QFileDialog.getSaveFileName(None, "Choose Image", "",
|
"All (*.*)")
|
print(self.saved_file)
|
|
|
def display_image(self):
|
pixmap = QPixmap(self.chosen_file[0])
|
self.ui.origin_img.setPixmap(pixmap)
|
# now read the image and show it via Image in other container
|
with Image.open(self.chosen_file[0]) as img:
|
image = np.asarray(img, dtype=np.uint8)
|
self.display_numpy_image(image)
|
|
|
def display_numpy_image(self, image):
|
height, width, channel = image.shape
|
print(height, width, channel)
|
|
bytes_per_line = channel * width # color
|
img_format = QImage.Format.Format_RGBX8888
|
if channel == 4:
|
img_format = QImage.Format.Format_ARGB32
|
elif channel == 3:
|
img_format = QImage.Format.Format_RGB888
|
qImg = QImage(image.data, width, height, bytes_per_line, img_format).rgbSwapped()
|
pixmap = QPixmap(qImg)
|
self.ui.result_img.setPixmap(pixmap)
|
|
|
|
|
def main():
|
app = QApplication(sys.argv)
|
window = MainWindow()
|
window.show()
|
|
sys.exit(app.exec())
|
pass
|
|
|
if __name__ == "__main__":
|
main()
|
