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

몬발켜 · 2024.04.30 12:37:03

<내가 chatGPT4랑 의논해서 만든 코드임... 다른 사람 코드는 전혀 들어있지 않음. 마음대로 쓰시라고 공개함.>

import cv2
import os
import numpy as np
import sys

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)]

# 각 파일을 하나씩 불러온다
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)

        

sys.exit()

번호	제목	글쓴이	작성일	조회	추천
설문	경제관념 부족해서 돈 막 쓸 것 같은 스타는?	운영자	24/05/13	-	-
2696187	이러니 살이 안빠지지..ㅇㅅㅇ [2]	헤르 미온느	05.16	24	0
2696186	빙산의 일각	발명도둑잡기	05.16	12	0
2696185	고죠 부활하니까 전세계트래픽 1위 ㄷㅅㄷ	♥냥덩배달왔습니당♥	05.16	38	0
2696184	지금 만드는 토이 프로젝트 딸깍 하면 인공지능이 해주는데	프갤러(121.186)	05.16	30	0
2696183	미래의 어느 역사학자들의 대화라는데	프갤러(211.241)	05.16	29	0
2696182	카시트 침대	발명도둑잡기	05.16	9	0
2696180	ㅎㅍㄴㅅ야 좋은 음악 추천해줄게 [4]	딱국(113.198)	05.16	33	0
2696179	백엔드 현실 연봉 얼마냐? [6]	ㅇㅇ(223.38)	05.16	158	0
2696178	오늘의 발명 실마리: 가짜 꼬리 항문 괄약근으로 조종하는 장치	발명도둑잡기	05.16	13	0
2696177	나는 영혼을 담는다구	ㅇㅇ(118.235)	05.16	12	0
2696176	프론트엔드 면접관 들어가는 애들 들어와봐 [7]	프갤러(221.150)	05.16	123	0
2696175	잡코리아에서 그냥 공고 확인도 안하고 체크하고 일괄지원했다	프갤러(112.150)	05.16	20	0
2696174	나님 주무시기전 소통합니당❤+ 질문받아양✨ [3]	♥냥덩배달왔습니당♥	05.16	29	0
2696173	요즘 코테 릿코드 하드까진 풀어야되더라 [3]	ㅇㅇ(210.178)	05.16	56	0
2696172	ai로 인해 개발자 미래 없음? [1]	ㅇㅇ(118.235)	05.16	68	0
2696171	나는 사실 찐따가 아니다 [2]	발명도둑잡기	05.16	37	0
2696170	차가운 도시.. 따뜻한 밤..	♥냥덩배달왔습니당♥	05.16	14	0
2696169	내부 고발자가 무서운 이유	ㅇㅇ(183.109)	05.16	40	0
2696168	싸이코패쓰 시니어 프로그래머 냥덩 이민 간다는데	발명도둑잡기	05.16	17	0
2696167	진짜 씨발씨발씨발 좆같다 [8]	프갤러(59.18)	05.16	101	0
2696166	복지부 털렸다	조커@두팩트수보(175.117)	05.16	20	0
2696165	애널의 수면영상	♥냥덩배달왔습니당♥	05.16	18	0
2696164	마이크로서비스들을 모노레포로 관리하면 문제가 뭐있음?	ㅇㅇ(223.38)	05.16	18	0
2696160	비전공 국비 맘 접었다 낼 마지막 면접임 ㅇㅇ [1]	ㅇㅇ(211.234)	05.16	68	0
2696159	취업 후기. 비전공 국비 신입 아니다. 경력자임. [2]	프갤러(59.16)	05.16	87	0
2696158	RxFramework 샘플 코드 및 시연 자료, 유체역학 코드 [5]	프갤러(121.172)	05.16	38	1
2696157	인의협 "尹 의료개혁 비현실적...'의료 민영화' 향해"	발명도둑잡기	05.16	11	0
2696156	RxStore 원본코드 [2]	프갤러(121.172)	05.16	42	1
2696155	찐따 탈출하는 법 알려준다	나트륨찡	05.16	26	0
2696153	나님 누엇어양.. [2]	♥냥덩배달왔습니당♥	05.16	28	0
2696152	중소기업 신입 면접, 코딩테스트 어느정도 수준이어야되는거임 .. ?? [3]	프갤러(112.150)	05.16	84	0
2696151	금융은 그냥 하나의 정치판이야 [2]	버거띠	05.16	55	0
2696150	범죄자집단 민주당 의료민영화 시동걸었네 ㄷㅅㄷ	♥냥덩배달왔습니당♥	05.16	27	0
2696149	진지하게 원하는거 다가지고 그렇게 살고싶다 씨발	ㅇㅇ(123.213)	05.16	28	0
2696148	난 업무핵심 걍 다 공유하고 정리해서 다 쏴줌 [1]	버거띠	05.16	40	0
2696146	[더 문] X 윤도현｜'착한사람 컴플렉스' 콜라보 MV	발명도둑잡기	05.16	9	0
2696145	내일은 불금이니	슈퍼막코더(116.64)	05.16	10	0
2696144	나도 손자병법 같은거 정독하면 찐따 벗어 날 수 있음? [2]	다칸마리	05.16	16	0
2696143	담배 한대 피구 주무실 준비..	♥냥덩배달왔습니당♥	05.16	15	0
2696142	c언어 공부하는데 이진법 [1]	ㅇㅇ(211.36)	05.16	29	0
2696141	프리 개발자들 사업자 안내고 하면 세금 3.3%만 내잖아	프갤러(114.204)	05.16	28	0
2696140	Cat vs dog	발명도둑잡기	05.16	11	0
2696139	낼은 정규직 시다해야된다	프갤러(118.235)	05.16	12	0
2696138	고지능자가 취업이 힘든 이유 [1]	프갤러(183.96)	05.16	96	1
2696137	12일차되니까 지치네	프갤러(175.119)	05.16	17	0
2696136	웹개발용 데탑 견적 짜봤어 [6]	프갤러(221.167)	05.16	69	0
2696135	애초에 cns sds 대기업 외주인데	프갤러(175.119)	05.16	27	0
2696134	나님 명작 만화 ㅊㅊ	♥냥덩배달왔습니당♥	05.16	12	0
2696132	코어 업무를 외주 주는 곳은 가지마라 [7]	포항의봄	05.16	72	0
2696131	주니어 하니까 주니어아이돌이랑 섹스하고 싶네 [4]	류류(121.160)	05.16	32	0

갤러리 검색

최근 방문

즐겨찾기

즐겨찾기 갤러리

프로그래밍 갤러리

머리말∙꼬리말

머리말∙꼬리말

색상 설정

갤러리 정보

스포일러 경고 설정

제목에서 경고

본문에서 경고

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

연관 갤러리

개념글 리스트

차단하기

[프로그래밍 갤러리]

갤러리 본문 영역

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

추천 비추천

댓글 영역

① NFT 발행

② NFT 구매

하단 갤러리 리스트 영역

왼쪽 컨텐츠 영역

갤러리 리스트 영역

페이지 이동

오른쪽 컨텐츠 영역

알림 설정

알림

실시간 베스트

뉴스

디시미디어

디시이슈

개념글[도시]

디시콘 리스트

디시콘

디시콘 검색결과(0)

인기 디시콘

지갑 연결