01_environment_basics.py

  1"""
  201. 환경 설정 및 기초
  3- OpenCV 설치 확인
  4- 버전 정보
  5- 기본 테스트 이미지 생성
  6"""
  7
  8import cv2
  9import numpy as np
 10import sys
 11
 12
 13def check_opencv_installation():
 14    """OpenCV 설치 및 버전 확인"""
 15    print("=" * 50)
 16    print("OpenCV 환경 확인")
 17    print("=" * 50)
 18
 19    # 버전 확인
 20    print(f"OpenCV 버전: {cv2.__version__}")
 21    print(f"Python 버전: {sys.version}")
 22    print(f"NumPy 버전: {np.__version__}")
 23
 24    # 빌드 정보
 25    print("\n[빌드 정보]")
 26    build_info = cv2.getBuildInformation()
 27    # 주요 정보만 출력
 28    for line in build_info.split('\n')[:20]:
 29        if line.strip():
 30            print(line)
 31
 32
 33def check_available_modules():
 34    """사용 가능한 모듈 확인"""
 35    print("\n" + "=" * 50)
 36    print("사용 가능한 주요 기능")
 37    print("=" * 50)
 38
 39    # SIFT 확인 (contrib 필요)
 40    try:
 41        sift = cv2.SIFT_create()
 42        print("SIFT: 사용 가능")
 43    except AttributeError:
 44        print("SIFT: 사용 불가 (opencv-contrib-python 필요)")
 45
 46    # ORB 확인
 47    try:
 48        orb = cv2.ORB_create()
 49        print("ORB: 사용 가능")
 50    except AttributeError:
 51        print("ORB: 사용 불가")
 52
 53    # DNN 확인
 54    try:
 55        net = cv2.dnn.readNet
 56        print("DNN 모듈: 사용 가능")
 57    except AttributeError:
 58        print("DNN 모듈: 사용 불가")
 59
 60    # Haar Cascade 확인
 61    cascade_path = cv2.data.haarcascades + 'haarcascade_frontalface_default.xml'
 62    print(f"Haar Cascade 경로: {cv2.data.haarcascades}")
 63
 64
 65def create_test_image():
 66    """테스트용 이미지 생성"""
 67    print("\n" + "=" * 50)
 68    print("테스트 이미지 생성")
 69    print("=" * 50)
 70
 71    # 컬러 이미지 생성 (400x400, BGR)
 72    img = np.zeros((400, 400, 3), dtype=np.uint8)
 73
 74    # 배경 그라데이션
 75    for i in range(400):
 76        img[i, :] = [i * 255 // 400, 100, 255 - i * 255 // 400]
 77
 78    # 도형 그리기
 79    # 사각형
 80    cv2.rectangle(img, (50, 50), (150, 150), (0, 255, 0), 2)
 81    cv2.rectangle(img, (60, 60), (140, 140), (0, 255, 0), -1)  # 채움
 82
 83    # 원
 84    cv2.circle(img, (300, 100), 50, (255, 0, 0), 2)
 85    cv2.circle(img, (300, 100), 30, (255, 0, 0), -1)
 86
 87    # 선
 88    cv2.line(img, (50, 250), (350, 250), (0, 0, 255), 3)
 89    cv2.line(img, (50, 280), (350, 320), (255, 255, 0), 2)
 90
 91    # 타원
 92    cv2.ellipse(img, (200, 350), (100, 30), 0, 0, 360, (255, 0, 255), 2)
 93
 94    # 텍스트
 95    cv2.putText(img, 'OpenCV Test', (100, 200),
 96                cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2)
 97
 98    # 저장
 99    cv2.imwrite('sample.jpg', img)
100    print("sample.jpg 생성 완료")
101
102    # 그레이스케일 이미지도 생성
103    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
104    cv2.imwrite('sample_gray.jpg', gray)
105    print("sample_gray.jpg 생성 완료")
106
107    return img
108
109
110def basic_operations_demo(img):
111    """기본 연산 데모"""
112    print("\n" + "=" * 50)
113    print("기본 연산 데모")
114    print("=" * 50)
115
116    # 이미지 속성
117    print(f"이미지 shape: {img.shape}")
118    print(f"이미지 dtype: {img.dtype}")
119    print(f"이미지 size: {img.size}")
120
121    # 픽셀 접근
122    pixel = img[100, 100]
123    print(f"픽셀 (100, 100) BGR 값: {pixel}")
124
125    # ROI (Region of Interest)
126    roi = img[50:150, 50:150]
127    print(f"ROI shape: {roi.shape}")
128
129
130def main():
131    """메인 함수"""
132    # 환경 확인
133    check_opencv_installation()
134    check_available_modules()
135
136    # 테스트 이미지 생성
137    img = create_test_image()
138
139    # 기본 연산 데모
140    basic_operations_demo(img)
141
142    # 이미지 표시 (GUI 환경에서)
143    print("\n[이미지 표시]")
144    print("이미지 창을 닫으려면 아무 키나 누르세요...")
145
146    try:
147        cv2.imshow('Test Image', img)
148        cv2.waitKey(0)
149        cv2.destroyAllWindows()
150    except Exception as e:
151        print(f"GUI 표시 불가: {e}")
152        print("Headless 환경에서는 cv2.imshow() 사용 불가")
153
154    print("\n환경 설정 확인 완료!")
155
156
157if __name__ == '__main__':
158    main()