본문 바로가기

728x90

pil

(2)
[이미지] CustomDataset [Python] - cv2, PIL 아직 모르는 게 많지만, 작성한 코드입니다. train 폴더 내에 label 이름별 폴더가 들어 있고, 그 내부에 이미지가 들어있는 구조입니다. test 폴더 같은 계층에 존재한다는 가정하에 향후 이용을 위해 train 폴더 상위 폴더에서 os.path.join으로 train을 더하고 os.walk로 내부 폴더와 파일명을 가져옵니다. for문을 돌려서 위치에 맞게 이미지 파일 경로와 라벨을 저장하고, 라벨은 학습을 위해 딕셔너리를 활용해 숫자로 변경합니다. 적절한 변형을 가한 후 window창을 띄워 확인합니다. import cv2 import os from PIL import Image import numpy as np from torch.utils.data import Dataset, DataLoad..
[이미지] 번호판 생성 [Python] - cv2, PIL 2006년 이전쯤 기준 차량 번호판 생성 도시와 문자 숫자를 랜덤으로 선택 높이, 너비 정보를 구하고 정보를 넣을 좌표 구하기 PIL 이미지로 문자를 이미지에 넣기 # 문자를 이미지에 넣기 위해 PIL 파일로 변환 과정 필요 # Image.fromarray()에 변경할 이미지를 입력 # 문자를 삽입하기 위해 ImageDraw.Draw()에 PIL 이미지로 변환된 파일을 입력 # draw.text()에 시작 좌표(가로, 세로), 문자, 색상, 폰트.. 등등을 입력 final_img = Image.fromarray(org_img) draw = ImageDraw.Draw(final_img) draw.text((450, 150), upper_str, (255,255,255), font=font_u) 번호판 10..