북스캔 파이썬 코드 퍼블릭 도메인 (2)

몬발켜 · 2024.04.30 12:39:57

<PyQT5를 이용해서 GUI를 추가함... 허접한 GUI이긴 하지만..>

import sys
import os
import cv2
import numpy as np
from PyQt5.QtWidgets import QApplication, QWidget, QPushButton, QVBoxLayout, QLabel, QMessageBox
from PyQt5.QtCore import QThread, pyqtSignal
from PyQt5.QtGui import QFont
import time



class ImageProcessingThread(QThread):
    update_signal = pyqtSignal(int, int)
    finished_signal = pyqtSignal()  # 작업 완료 시그널

    def run(self):
        try:
            folder_path = "C:/input"
            file_list = os.listdir(folder_path)
            image_extensions = [".jpg", ".jpeg", ".png"]
            image_files = [file for file in file_list if any(file.lower().endswith(ext) for ext in image_extensions)]

            total_files = len(image_files)
            processed_files = 0

            # 각 파일을 하나씩 불러온다
            for image_name in image_files:

                # 이미지 파일을 불러온다
                image = cv2.imread("C:\\input\\{}".format(image_name))

                # 그레이 스케일 및 이진화
                gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
                _, binary = cv2.threshold(gray, 127, 255, cv2.THRESH_BINARY | cv2.THRESH_OTSU )

                # 컨투어 찾기 
                contours, _ = cv2.findContours(binary, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

                # 가장 큰 컨투어 찾기
                largest_contour = max(contours, key=cv2.contourArea)

                # 가장 큰 컨투어에 불필요한 점/점들이 포함된 경우가 종종 발생하기 때문에 제거할 필요가 있다
                # 새로운 이미지 생성 (검은색 배경)
                new_image = np.zeros_like(binary)

                # 가장 큰 컨투어 중에서 면적이 가장 큰 폐곡선 찾기
                largest_closed_contour = max([c for c in contours if cv2.contourArea(c) > 0], key=cv2.contourArea)

                # 면적이 가장 큰 폐곡선 안에 있는 흰색 점들만 남기고 나머지는 삭제하여 새 이미지에 저장
                cv2.drawContours(new_image, [largest_closed_contour], -1, (255), thickness=cv2.FILLED)

                # 컨투어를 둘러싼 가장 작은 사각형 찾기
                min_rect = cv2.minAreaRect(largest_closed_contour)

                # minAreaRect에서 반환한 4개 점은 실수값을 갖으므로, 사용하려면 정수로 변환한다
                box_points = cv2.boxPoints(min_rect)
                box_points = np.intp(box_points)

                # 4개 점을 좌표값에 따라서 번호를 부여한다
                # 원래의 의도는 좌상, 우상, 좌하, 우하 순서로 P1, P2, P3, P4를 부여하려고 하였다
                # 그러나 프로그래머의 일반적인 코드 규칙에 따라서 점의 번호를 부여하기로 생각을 바꿨다.
                # The order will be P1 (top-left), P2 (top-right), P3 (bottom-left), P4 (bottom-right)
                box_points = sorted(box_points, key=lambda x: (x[1], x[0]))
                if box_points[0][0] > box_points[1][0]:
                    box_points[0], box_points[1] = box_points[1], box_points[0]
                if box_points[2][0] > box_points[3][0]:
                    box_points[2], box_points[3] = box_points[3], box_points[2]

                # 프로그래머의 일반적인 코드 부여 방법에 따라서 점의 번호를 정했다
                P1 = box_points[0]
                P2 = box_points[1]
                P3 = box_points[2]
                P4 = box_points[3]

                if P1[0] < P2[0]: # 기울기가 양수일 때(왼쪽으로 기울었다:시계방향으로 회전해야 한다)
                    x1, y1 = P1[0], P1[1]
                    x2, y2 = P2[0], P2[1]

                    # 기울기 slope를 계산한다. 
                    delta_x = x2 - x1
                    delta_y = y2 - y1
                    slope = delta_y / delta_x if delta_x != 0 else None

                    # 기울기에 따라서 회전할 각과 방향을 계산한다
                    # Calculate the angle in radians and then convert to degrees
                    # The angle must be negative for a clockwise rotation
                    angle_of_rotation = -np.degrees(np.arctan(slope))

                    # 시계방향으로 angle만큼 회전하고, 변수에 저장한다
                    # center에 들어갈 점 P의 좌표는 데이터형을 변환해야 한다
                    center = (int(P1[0]), int(P1[1])) 
                    angle = -angle_of_rotation
                    M = cv2.getRotationMatrix2D(center, angle, 1)
                    rotated_image = cv2.warpAffine(image, M, (image.shape[1], image.shape[0]))

                    # 회전한 이미지를 저장하지 않고 변수를 바로 사용하여 다음 단계의 작업을 진행한다 
                    #그레이 스케일 및 이진화
                    gray1 = cv2.cvtColor(rotated_image, cv2.COLOR_BGR2GRAY)
                    _, binary1 = cv2.threshold(gray1, 127, 255, cv2.THRESH_BINARY | cv2.THRESH_OTSU )

                    
                    # 회전한 이미지의 컨투어 찾기
                    contours1, _ = cv2.findContours(binary1, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

                    # 회전한 이미지의 가장 큰 컨투어 찾기
                    largest_contour1 = max(contours1, key=cv2.contourArea)

                    # 새로운 이미지 생성 (검은색 배경)
                    new_image1 = np.zeros_like(binary1)

                    # 가장 큰 컨투어에 불필요한 점/점들이 포함된 경우가 종종 발생하기 때문에 제거할 필요가 있다
                    # 회전한 이미지의 가장 큰 컨투어 중에서 면적이 가장 큰 폐곡선 찾기
                    largest_closed_contour1 = max([c for c in contours1 if cv2.contourArea(c) > 0], key=cv2.contourArea)

                    # 면적이 가장 큰 폐곡선 안에 있는 흰색 점들만 남기고 나머지는 삭제하여 새 이미지에 저장
                    cv2.drawContours(new_image1, [largest_closed_contour1], -1, (255), thickness=cv2.FILLED)

                    # 회전한 이미지의 컨투어를 포함하는 가장 작은 사각형을 찾는다
                    min_rect1 = cv2.minAreaRect(largest_closed_contour1)

                    # minAreaRect에서 반환한 4개 점은 실수값을 갖으므로, 사용하려면 정수로 변환한다
                    # Convert it to box points (four points)
                    box_points1 = cv2.boxPoints(min_rect1)
                    box_points1 = np.intp(box_points1)

                    # # 4개 점을 좌표값에 따라서 번호를 부여한다
                    # 원래의 의도는 좌상, 우상, 좌하, 우하 순서로 P1, P2, P3, P4를 부여하려고 하였다
                    # 그러나 프로그래머의 일반적인 코드 규칙에 따라서 점의 번호를 부여하기로 생각을 바꿨다.
                    # The order will be P1 (top-left), P2 (top-right), P3 (bottom-left), P4 (bottom-right)
                    box_points1 = sorted(box_points1, key=lambda x: (x[1], x[0]))
                    if box_points1[0][0] > box_points1[1][0]:
                        box_points1[0], box_points1[1] = box_points1[1], box_points1[0]
                    if box_points1[2][0] > box_points1[3][0]:
                        box_points1[2], box_points1[3] = box_points1[3], box_points1[2]

                    # 프로그래머의 일반적인 코드 부여 방법에 따라서 점의 번호를 정했다
                    # 점의 번호를 앞의 점과 다르게 해서 혼동을 피했다    
                    P11 = box_points1[0]
                    P22 = box_points1[1]
                    P33 = box_points1[2]
                    P44 = box_points1[3]
                    
                    # 4개 점의 좌표를 이용하여 사각형 영역을 택하여 새 그림 파일로 저장한다
                    k = rotated_image[P11[1]:P33[1], P11[0]:P22[0]]

                    cv2.imwrite("C:\\output\\{}".format(image_name), k)

                else: # 기울기가 양수가 아닐 때(오른쪽으로 기울었다:반시계방향으로 회전해야 한다)

                    x1, y1 = P1[0], P1[1]
                    x2, y2 = P2[0], P2[1]

                    # 기울기를 계산한다
                    delta_x = x2 - x1
                    delta_y = y2 - y1
                    slope = delta_y / delta_x if delta_x != 0 else None

                    # 기울기에 따라서 회전할 각과 방향을 계산한다
                    # If the slope is zero (horizontal line), we do not need to rotate.
                    # If the slope is negative or undefined (vertical line), we rotate counterclockwise.
                    # The angle must be positive for a counterclockwise rotation
                    angle_of_rotation = np.degrees(np.arctan(-slope)) if slope is not None else 90

                    # 반시계방향으로 angle만큼 회전하여 변수에 저장한다
                    # center에 들어갈 점 P의 좌표는 데이터형을 변환해야 한다
                    center = (int(P1[0]), int(P1[1])) 
                    angle = angle_of_rotation
                    M = cv2.getRotationMatrix2D(center, angle, 1)
                    rotated_image = cv2.warpAffine(image, M, (image.shape[1], image.shape[0]))

                    # 회전한 이미지를 저장하지 않고 변수를 바로 사용하여 다음 단계의 작업을 진행한다  
                    #그레이 스케일 및 이진화
                    gray1 = cv2.cvtColor(rotated_image, cv2.COLOR_BGR2GRAY)
                    _, binary1 = cv2.threshold(gray1, 127, 255, cv2.THRESH_BINARY | cv2.THRESH_OTSU )

                    # 회전한 이미지의 컨투어 찾기
                    contours1, _ = cv2.findContours(binary1, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

                    # 회전한 이미지의 가장 큰 컨투어 찾기
                    largest_contour1 = max(contours1, key=cv2.contourArea)

                    # 새로운 이미지 생성 (검은색 배경)
                    new_image1 = np.zeros_like(binary1)

                    # 회전한 이미지의 가장 큰 컨투어 중에서 면적이 가장 큰 폐곡선 찾기
                    largest_closed_contour1 = max([c for c in contours1 if cv2.contourArea(c) > 0], key=cv2.contourArea)

                    # 면적이 가장 큰 폐곡선 안에 있는 흰색 점들만 남기고 나머지는 삭제하여 새 이미지에 저장
                    cv2.drawContours(new_image1, [largest_closed_contour1], -1, (255), thickness=cv2.FILLED)

                    # 회전한 이미지의 컨투어를 포함하는 가장 작은 사각형을 찾는다
                    min_rect1 = cv2.minAreaRect(largest_closed_contour1)

                    # minAreaRect에서 반환한 4개 점은 실수값을 갖으므로, 사용하려면 정수로 변환한다
                    # Convert it to box points (four points)
                    box_points1 = cv2.boxPoints(min_rect1)
                    box_points1 = np.intp(box_points1)

                    # 4개 점을 좌표값에 따라서 번호를 부여한다
                    # 원래의 의도는 좌상, 우상, 좌하, 우하 순서로 P1, P2, P3, P4를 부여하려고 하였다
                    # 그러나 프로그래머의 일반적인 코드 규칙에 따라서 점의 번호를 부여하기로 생각을 바꿨다.
                    # The order will be P1 (top-left), P2 (top-right), P3 (bottom-left), P4 (bottom-right)
                    box_points1 = sorted(box_points1, key=lambda x: (x[1], x[0]))
                    if box_points1[0][0] > box_points1[1][0]:
                        box_points1[0], box_points1[1] = box_points1[1], box_points1[0]
                    if box_points1[2][0] > box_points1[3][0]:
                        box_points1[2], box_points1[3] = box_points1[3], box_points1[2]

                    # 프로그래머의 일반적인 코드 부여 방법에 따라서 점의 번호를 정했다
                    # 점의 번호를 앞의 점과 다르게 해서 혼동을 피했다   
                    P11 = box_points1[0]
                    P22 = box_points1[1]
                    P33 = box_points1[2]
                    P44 = box_points1[3]
                    
                    # 4개 점의 좌표를 이용하여 사각형 영역을 택하여 새 그림 파일로 저장한다
                    k = rotated_image[P11[1]:P33[1], P11[0]:P22[0]]

                    cv2.imwrite("C:\\output\\{}".format(image_name), k)
            


                processed_files += 1
                self.update_signal.emit(total_files, processed_files)

                time.sleep(0.1)  # 짧은 지연 추가


        # 모든 작업 완료 후 신호 발생 전에 짧은 지연을 추가
            self.finished_signal.emit()
            time.sleep(0.2)
        except Exception as e:
           print("Error during image processing:", e)
           self.finished_signal.emit()


class ImageProcessorGUI(QWidget):
    def __init__(self):
        super().__init__()

        self.initUI()

    def initUI(self):
        self.setWindowTitle('Dadong')
        self.setFixedSize(500, 300)  # 가로 500, 세로 300 크기로 고정

        self.layout = QVBoxLayout()

        # QFont 객체 생성
        font = QFont()
        font.setPointSize(font.pointSize() * 2)  # 현재 폰트 크기의 2배로 설정

      
        # 실행 버튼
        self.startButton = QPushButton('실행')
        self.startButton.setFont(font)  # 폰트 적용
        self.startButton.clicked.connect(self.startProcessing)
        self.layout.addWidget(self.startButton)
        
        # 전체 파일 수를 표시하는 레이블
        self.totalFilesLabel = QLabel('input total files: 0')
        self.totalFilesLabel.setFont(font)  # 폰트 적용
        self.layout.addWidget(self.totalFilesLabel)

        # 처리된 파일 수를 표시하는 레이블
        self.processedFilesLabel = QLabel('output processed files: 0')
        self.processedFilesLabel.setFont(font)  # 폰트 적용
        self.layout.addWidget(self.processedFilesLabel)

        # '설명' 버튼 추가
        self.descriptionButton = QPushButton('설명')
        self.descriptionButton.setFont(font)  # 폰트 적용
        self.descriptionButton.clicked.connect(self.showDescription)
        self.layout.addWidget(self.descriptionButton)

        self.setLayout(self.layout)

        self.thread = ImageProcessingThread()
        self.thread.update_signal.connect(self.updateStatus)
        self.thread.finished_signal.connect(self.closeProgram)  # 작업 완료 시그널 연결

    def showDescription(self):
        description = (
            "'C:의 input 폴더에 사진 파일을 저장하세요 \n\n"
            "파일 이름에 한글이 있으면 처리가 중단됩니다 \n\n"
            "C:의 output 폴더에 책 파일들이 저장됩니다 \n\n"
            "처리가 완료되면 프로그램이 자동 종료됩니다 \n\n"
            
        )
        QMessageBox.information(self, "설명", description)         

    def startProcessing(self):
        self.thread.start()

    def closeProgram(self):
        if self.thread.isRunning():  # 스레드가 실행 중인지 확인
            self.thread.quit()  # 스레드에 종료 요청
            self.thread.wait(1000)  # 스레드가 완전히 종료될 때까지 기다림
        self.close()  # 이후 GUI 종료

    # 상태 업데이트 메서드
    def updateStatus(self, total, processed):
        self.totalFilesLabel.setText(f'input total files: {total}')
        self.processedFilesLabel.setText(f'output processed files: {processed}')    



def main():
    app = QApplication(sys.argv)
    ex = ImageProcessorGUI()
    ex.show()
    sys.exit(app.exec_())

if __name__ == '__main__':
    main()

번호	제목	글쓴이	작성일	조회	추천
설문	현역으로 군대 안 간게 의아한 스타는?	운영자	25/06/30	-	-
AD	휴대폰 바꿀까? 특가 구매 찬스!	운영자	25/07/02	-	-
공지	프로그래밍 갤러리 이용 안내 [88]	운영자	20.09.28	45135	65
2869440	저녘 ㅇㅅㅇ	류류(118.235)	20:32	0	0
2869438	업뎃) 동일 안전성 목표 하의 코드 비교 - Rust vs. Ada/..	루비	20:05	15	0
2869437	7 [1]	ㅇㅇ(106.101)	20:02	13	0
2869435	나님 리짜이밍 덕분에 집값 폭등즁 ㅎㅅㅎ	♥냥덩이♥	19:59	14	0
2869431	냉정하게 23살이면 아직 기회있다고 보는게 맞는 나이임?? [2]	ㅇㅇ(223.38)	19:44	23	0
2869429	미국 스타일 여성분 마주쳤다 [2]	헬마스터	19:41	23	0
2869427	ㅁㅌ겠네.. 러빠들 파닥파닥 날뛰면서 개난리칠거 예상된다. [1]	루비	19:35	13	0
2869424	그러니까 무조건 러스트만을 찬양/옹호할게 아니라	루비	19:23	13	1
2869423	❤✨☀⭐나님 시작합니당⭐☀✨❤	♥냥덩이♥	19:23	11	0
2869422	부록 B: 동일 안전성 목표 하의 코드 비교 - Rust vs. Ada/ [2]	루비	19:19	10	0
2869420	러스트 빠 주장 분석: C언어도 stdio를 못 쓰는가?	루비	19:16	14	1
2869418	c++ 집에있는 책 펴봤느데 c++11기반 이네 [1]	프갤러(114.205)	19:06	23	0
2869417	새끼길냥덩 주울깡?	♥냥덩이♥	19:04	20	0
2869416	CPP 배울거면 그나마 스콧 메이어 책 이펙티브 모던C++봐야하는데 [1]	ㅆㅇㅆ(124.216)	19:04	28	0
2869415	c는 임베디드에서 stdio쓸 수 있는줄 아나봄	프갤러(42.22)	19:03	19	0
2869414	백악관이 rust ada 등 메모리안전 언어를 추천하는 가운데 kyle이 [1]	발명도둑잡기(118.216)	19:01	23	0
2869413	c와 런타임라니브러리 합친 크기와 동일 기능하는 러스트 프로그램 실행파일	발명도둑잡기(118.216)	18:59	14	0
2869412	임베디드 시스템 및 리눅스 커널 적용의 실제적 제약 (바이너리, ABI	루비	18:58	18	0
2869411	내일 내가 만든 임베디드 시연하러 가야함 [1]	프갤러(211.234)	18:56	27	0
2869410	CPP는 책으로 배우면 잘못 입문한다. [6]	ㅆㅇㅆ(124.216)	18:49	56	0
2869409	임베디드에서 동적링킹 쓰나 [5]	발명도둑잡기(118.216)	18:49	37	0
2869408	부록 A: ‘합리적 비판’에 대한 ‘비이성적 반응’ 사례 연구	루비	18:49	12	0
2869407	안되겠다. 자짤을 톡시한 크랩랭으로 복귀해야겠다.	프갤러(42.27)	18:49	11	0
2869406	나님 왤케 특별하실깡?	♥냥덩이♥	18:48	14	0
2869405	동적 링크는 시스템에 미리 그 라이브러리가 통째로 설치되어 있어야돼 [1]	프갤러(42.27)	18:46	19	0
2869404	행님들 C++입문할라는데 입문서 + 코딩문제집 동시에 보는거어 [2]	프갤러(123.142)	18:42	16	0
2869403	러빠하는 얘기보면 '동정 링킹'의 개념을 아예 모르는거 같음. ㅎㅎ [1]	루비	18:28	26	0
2869402	전공자 LG 유플러스 부트캠프 어떨까요?? [3]	프갤러(218.50)	18:27	33	0
2869401	보석아. 너는 지금 동적 링크가 어떻게 동작하는지도 모르고 있는거야. [1]	프갤러(27.177)	18:25	35	0
2869400	내가 llm 조련하는거 보고가~~[위 러빠 얘기도 넣어 업뎃함]	루비	18:24	21	1
2869399	그 보석새끼 또 임베디드 바이너리 크기 타령하네 ㅋㅋ	프갤러(223.33)	18:21	21	0
2869398	냥덩너무조와? [3]	♥냥덩이♥	18:04	33	0
2869397	네이버 카페가 안들어가짐 프갤을 잘 되는데, [1]	넥도리아(121.139)	18:04	14	0
2869396	ㅆㅇㅆ아 그건 너 말이 맞다. [2]	루비	17:59	46	1
2869395	지금 netstat, tasklist 결과 [2]	발명도둑잡기(118.216)	17:51	40	0
2869394	커뮤니티 사이트 회원가입 빡세게 해놓는 이유...txt	ㅇㅇ	17:48	21	0
2869393	나는 근데 루비보면서 LLM의 무서운점을 느낌. [4]	ㅆㅇㅆ(124.216)	17:47	46	2
2869392	ㅆㅇㅆ 고맙다. 어제 마니 까줘서 보강했어 [2]	루비	17:43	33	1
2869391	애들아 메모리쪽 공부해 [6]	프갤러(106.101)	17:31	89	1
2869390	갤에 러까 글 진지하게 읽는 사람 있음? [3]	프갤러(218.154)	17:30	37	0
2869389	추억의 만찐두빵⭐+	♥냥덩이♥	17:29	19	0
2869388	러스트 혹세무민 = 곡학아세 + 오만 + 기만	루비	17:29	15	1
2869387	에필로그: 러스트 신화는 막을 내린다	루비	17:24	18	0
2869386	지금까지 나 신고한 새끼들 나중에 싸그리 신고해서 추적한다 [1]	발명도둑잡기(118.216)	17:23	26	0
2869385	배달의민족 배달료 중앙값 표시 항의	발명도둑잡기(39.7)	17:20	23	0
2869384	제5장: 러스트의 설계, 혁신인가 결함인가?	루비	17:18	30	0
2869383	❤✨☀⭐나님 시작합니당⭐☀✨❤	♥냥덩이♥	17:16	15	0
2869382	1400줄 추가 1000줄 삭제하고 돌렸는데	←Backspace	17:14	34	0
2869381	가끔 코딩이 뭘까 싶기도하다. 코딩이라는 무형적 지식은 [6]	ㅆㅇㅆ(124.216)	17:11	57	0
뉴스	‘풋풋한 로맨스‘ 이채연X시온, 슬픈 이별로 안방극장 울렸다! 사랑∙꿈 이룰 수 있을까?	디시트렌드	10:00

최근 방문

즐겨찾기

즐겨찾기 갤러리

이미지 올리기 이용안내

갤러리 이슈박스, 최근방문 갤러리

연관 갤러리

개념글 리스트

차단하기

[프로그래밍 갤러리]

갤러리 본문 영역

추천 비추천

댓글 영역

① NFT 발행

② NFT 구매

하단 갤러리 리스트 영역

왼쪽 컨텐츠 영역

갤러리 리스트 영역

페이지 이동

오른쪽 컨텐츠 영역

알림 설정

알림

디시콘 리스트

디시콘

디시콘 검색결과(0)

인기 디시콘