import sys
from distutils.dir_util import copy_tree
from PyQt5.QtCore import QProcess, Qt
from PyQt5.QtGui import QPixmap
from PyQt5.QtWidgets import *
from PyQt5 import uic
from PyQt5.QtWidgets import QApplication, QMainWindow
import numpy as np
import cv2
import tensorflow as tf
form_class = uic.loadUiType("cnn_pj.ui")[0]
class WindowClass(QMainWindow, form_class):
def __init__(self):
super().__init__()
self.setupUi(self)
# ----------------------------↓↓↓변수↓↓↓-----------------------------------
# 모델
self.model_path = ""
# 데이터셋 파일 저장 리스트
self.dataset_list_dir = ""
# 데이터셋 적용폴더 저장 리스트
self.dataset_dir = ""
# 결과폴더 절대경로
self.result_dir = ""
# 텍스트 데이터 모든파일의 절대경로
self.text_all_list = []
# 그림 데이터 모든파일의 절대경로
self.img_all_list = []
# 화면에 표시된 데이터셋 파일과 결과파일의 절대값 주소
self.original_img = ""
self.result_img_img = ""
# treeView 목록 파일을 클릭하면 그 파일의 절대값을 저장
self.tree_View_file_click = ""
# 마진조절
self.margin = 15
# area조절
self.area = 1000
# ----------------------------↑↑↑변수↑↑↑-----------------------------------
# ----------------------------↓↓↓버튼↓↓↓-----------------------------------
# 모델/데이터셋 불러오기 버튼
self.pushButton_2.clicked.connect(self.datasetDirSelect)
# 결과폴더 선택버튼
self.pushButton_7.clicked.connect(self.resultDir)
# 리셋버튼
self.pushButton_5.clicked.connect(self.reset)
# 마진조정버튼
self.pushButton_margin.clicked.connect(self.marginC)
# area조정버튼
self.pushButton_area.clicked.connect(self.areaC)
# 결과확인 버튼 test
self.pushButton_result.clicked.connect(self.numberDetection)
# ----------------------------↑↑↑버튼↑↑↑-----------------------------------
# ----------------------------↓↓↓함수들↓↓↓--- --------------------------------
# 모델위치 선택 함수
def modelSelect(self):
file_path = QFileDialog.getOpenFileName(self, 'Attach File')[0]
self.model_path = file_path
# 데이터셋이 들어있는 폴더를 선택하는 함수
def datasetDirSelect(self):
file_path = QFileDialog.getExistingDirectory(self, "select Directory")
self.dataset_list_dir = file_path
self.model_file_system = QFileSystemModel()
self.model_file_system.setRootPath(file_path)
self.model_file_system.setReadOnly(False)
self.treeView.setModel(self.model_file_system)
self.treeView.setRootIndex(self.model_file_system.index(file_path))
self.treeView.doubleClicked.connect(lambda index: self.treeViewDoubleClicked1(index))
self.treeView.setDragEnabled(True)
self.treeView.setColumnWidth(0, 300)
# treeView 더블클릭 시 절대값 추출함수
def treeViewDoubleClicked1(self, index):
self.tree_View_file_click = self.model_file_system.filePath(index)
self.original_img = self.tree_View_file_click
pixmap = QPixmap(self.tree_View_file_click)
pixmap = pixmap.scaled(self.label.size(), aspectRatioMode=True)
self.label.setPixmap(pixmap)
def treeViewDoubleClicked2(self, index):
self.tree_View_file_click = self.model_file_system.filePath(index)
self.result_img = self.tree_View_file_click
pixmap = QPixmap(self.tree_View_file_click)
pixmap = pixmap.scaled(self.label_2.size(), aspectRatioMode=True)
self.label_2.setPixmap(pixmap)
# 실행버튼 함수
def runFile(self):
copy_tree(self.dataset_list_dir, self.dataset_dir)
file_path = self.model_path
process = QProcess(self)
process.start('python', [file_path])
# 데이터셋을 적용할 폴더 선택 함수
def datasetDir(self):
directory_path = QFileDialog.getExistingDirectory(self, "select Directory")
self.dataset_dir = directory_path
# 결과폴더 선택
def resultDir(self):
file_path = QFileDialog.getExistingDirectory(self, "select Directory")
self.result_dir = file_path
print(self.result_dir)
self.model_file_system = QFileSystemModel()
self.model_file_system.setRootPath(file_path)
self.model_file_system.setReadOnly(False)
self.treeView_2.setModel(self.model_file_system)
self.treeView_2.setRootIndex(self.model_file_system.index(file_path))
self.treeView_2.doubleClicked.connect(lambda index: self.treeViewDoubleClicked2(index))
self.treeView_2.setDragEnabled(True)
self.treeView_2.setColumnWidth(0, 300)
def reset(self):
self.label.clear()
self.label_2.clear()
self.lineEdit_area.clear()
self.lineEdit_margin.clear()
self.margin = 15
self.area = 1000
def marginC(self):
text = self.lineEdit_margin.text()
self.margin = text
def areaC(self):
text = self.lineEdit_area.text()
self.area = text
def numberDetection(self, pyqtExceptionHandler=None):
src = cv2.imread(self.original_img)
gray = cv2.cvtColor(src, cv2.COLOR_BGR2GRAY)
gray = 255 - gray
kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (3, 3))
gray = cv2.dilate(gray, kernel, iterations=1)
ret_val, binImg = cv2.threshold(gray, 100, 255, cv2.THRESH_OTSU)
ret_val, gray = cv2.threshold(gray, 94, 255, cv2.THRESH_TOZERO)
n_blob, labelImg, stats, centroid = cv2.connectedComponentsWithStats(binImg)
model = tf.keras.models.load_model("cnn_model.h5")
result = []
margin = self.margin
try:
for i in range(1, n_blob):
x, y, w, h, area = stats[i]
if area > (self.area):
x -= margin
y -= margin
w += margin * 2
h += margin * 2
cv2.rectangle(src, (x, y, w, h), (255, 0, 255), thickness=2)
crop = cv2.getStructuringElement(cv2.MORPH_RECT, (10, 10))
crop = gray[y:y + h, x:x + w].copy()
crop2 = cv2.getStructuringElement(cv2.MORPH_RECT, (10, 10))
crop2 = src[y:y + h, x:x + w].copy()
crop = cv2.resize(crop, (28, 28), interpolation=cv2.INTER_AREA)
crop2 = cv2.resize(crop2, (28, 28), interpolation=cv2.INTER_AREA)
input_x = np.expand_dims(crop, axis=0)
output_y = model.predict(input_x)
ans = output_y.argmax()
conf = output_y[0, ans]
show_str = str(ans)
print(show_str)
result.append(show_str)
cv2.putText(src, show_str, (x, y), cv2.FONT_HERSHEY_PLAIN, 2, (0, 255, 0), 2)
# cv2.imwrite(f'{self.result_dir}/{i - 1}result_{show_str}.png', crop)
cv2.imwrite(f'{self.result_dir}/{i - 1}result_{show_str}.png', crop2)
except Exception as e:
print(self, f'Error : {str(e)}')
cv2.imwrite(f'{self.result_dir}/{result}.png', src)
# 오류처리
sys.excepthook = pyqtExceptionHandler
# cv2.imshow("crop ing", crop)
# cv2.imshow("gray img", gray)
cv2.imshow("original img", src)
# cv2.imshow("binary img", binImg)
cv2.waitKey(0)
cv2.destroyAllWindows()
# ----------------------------↑↑↑함수들↑↑↑-----------------------------------
if __name__ == "__main__" :
app = QApplication(sys.argv)
myWindow = WindowClass()
myWindow.show()
app.exec_()
Python 을통해 손글씨 이미지에서 숫자를 검출하는 하는 과정을 PyQt 를 통해 GUI로 제작
1. 이미지가 들어있는 폴더를 선택하면 오른쪽의 탐색창에 파일들이 표시되며, 탐색창에서 검출 할 이미지를 선택 2. 숫자검출 버튼을 누르면 검출 시작, 오른쪽 탐색창의 결과텝에서 파일들을 확인 할 수 있음. 3. 검출이 잘 되지 않을 떄에는 오른쪽 아래의 margin과 area를 조절해서 검출 결과를 더 좋게 수정 할 수 있음.
안녕하세요! 😊 오늘은 Python을 활용해서 PyQt를 사용하는 방법에 대해 알아보려고 해요. PyQt는 Python에서 GUI(그래픽 사용자 인터페이스) 애플리케이션을 만들 수 있게 도와주는 강력한 프레임워크예요. 그럼 시작해볼까요?
PyQt란 무엇인가? PyQt는 Qt 프레임워크를 기반으로 한 Python 바인딩으로, 다양한 플랫폼에서 GUI 애플리케이션을 개발할 수 있도록 해줘요. Qt는 C++로 작성된 라이브러리로, 다양한 기능을 제공하며, PyQt는 이를 Python에서 사용할 수 있게 해주는 역할을 해요. PyQt를 사용하면 복잡한 GUI를 쉽게 만들 수 있어요.
PyQt 설치하기 PyQt를 사용하기 위해서는 먼저 설치가 필요해요. PyQt5를 설치하는 방법은 다음과 같아요:
파이썬 설치 : 먼저 Python이 설치되어 있어야 해요. Python 공식 웹사이트에서 다운로드할 수 있어요.
PyQt5 설치 : 터미널이나 명령 프롬프트를 열고 다음 명령어를 입력해 주세요. bash pip install PyQt5
설치 확인 : 설치가 완료되면, Python 인터프리터를 열고 다음 코드를 입력해 보세요. python import PyQt5 print(PyQt5.version)
버전 정보가 출력되면 성공적으로 설치된 거예요!
기본 GUI 애플리케이션 만들기 이제 간단한 GUI 애플리케이션을 만들어 볼게요. 아래 코드를 참고해 주세요.
python import sys from PyQt5.QtWidgets import QApplication, QLabel
이 코드를 실행하면 "안녕하세요, PyQt5!"라는 텍스트가 있는 창이 나타나요. 정말 간단하죠? 😄
PyQt의 주요 위젯 소개 PyQt에서는 다양한 위젯을 제공해요. 몇 가지 주요 위젯을 소개할게요:
QPushButton : 버튼을 만들 수 있어요. QLineEdit : 텍스트 입력 필드를 만들 수 있어요. QLabel : 텍스트나 이미지를 표시할 수 있어요. QComboBox : 드롭다운 목록을 만들 수 있어요. QCheckBox : 체크박스를 만들 수 있어요. 이 외에도 많은 위젯이 있으니, 필요에 따라 적절한 위젯을 선택해 사용하면 돼요.
이벤트 처리 및 시그널/슬롯 PyQt의 가장 큰 장점 중 하나는 이벤트 처리 시스템이에요. 사용자가 버튼을 클릭하거나 텍스트를 입력할 때 발생하는 이벤트를 처리할 수 있어요. 이를 위해 시그널과 슬롯 개념을 사용해요.
안녕하세요! 오늘은 Python을 활용해서 손글씨를 검출하는 방법에 대해 알아보려고 해요. 손글씨 인식은 요즘 인공지능 기술이 발전하면서 많은 관심을 받고 있는 분야 중 하나인데요, 특히 TensorFlow와 NumPy를 사용하면 손글씨 이미지를 쉽게 처리할 수 있어요. 그럼 시작해볼까요?
1. 손글씨 검출이란? 손글씨 검출은 사람이 쓴 글씨를 컴퓨터가 인식하는 기술이에요. 이 기술은 주로 이미지 처리와 머신러닝을 기반으로 하며, 다양한 분야에서 활용되고 있어요. 예를 들어, 우편물 분류, 자동 채점 시스템, 그리고 개인화된 학습 도구 등에서 사용되죠.
2. 필요한 라이브러리 설치하기 손글씨 검출을 위해서는 몇 가지 라이브러리를 설치해야 해요. 가장 먼저 TensorFlow와 NumPy를 설치해볼게요. 아래의 명령어를 터미널에 입력해 주세요.
bash pip install tensorflow numpy
이 외에도 OpenCV를 사용하면 이미지 전처리에 유용하니, 필요하다면 설치해 주세요.
bash pip install opencv-python
3. 데이터셋 준비하기 손글씨 인식을 위해서는 데이터셋이 필요해요. MNIST 데이터셋은 손글씨 숫자 이미지로 구성되어 있어, 손글씨 인식 모델을 학습하는 데 많이 사용돼요. TensorFlow에서는 이 데이터셋을 쉽게 불러올 수 있어요.
이렇게 하면 모델의 정확도를 확인할 수 있어요. 만약 직접 손글씨 이미지를 검출하고 싶다면, OpenCV를 사용해 이미지를 불러오고 전처리한 후 모델에 입력하면 돼요.
7. 결과 확인하기 모델의 예측 결과를 확인해보는 것도 중요해요. 아래의 코드를 사용해 예측 결과를 시각화해볼 수 있어요.
python import matplotlib.pyplot as plt
predictions = model.predict(x_test)
for i in range(5): plt.imshow(x_test[i], cmap='gray') plt.title(f'예측: ') plt.show()
이렇게 하면 예측한 숫자와 실제 숫자를 비교할 수 있어요. 시각적으로 확인하는 것이 많은 도움이 될 거예요.
8. 마무리 및 추가 자료 오늘은 Python과 TensorFlow를 사용하여 손글씨를 검출하는 방법에 대해 알아보았어요. 손글씨 인식은 다양한 분야에서 활용될 수 있는 유용한 기술이니, 여러분도 한번 도전해보세요! 추가적으로 더 깊이 있는 학습을 원하신다면, TensorFlow 공식 문서나 관련 블로그를 참고하시면 좋을 것 같아요.
이상으로 손글씨 검출에 대한 포스팅을 마치겠습니다. 궁금한 점이 있다면 댓글로 남겨주세요!
데이터셋 준비하기 YOLOv5를 사용하기 위해서는 객체 탐지에 사용할 데이터셋이 필요해요. 일반적으로는 이미지와 해당 이미지에 대한 라벨 파일이 있어야 해요. 예를 들어, 고양이와 개를 탐지하고 싶다면 각각의 이미지를 준비하고, 해당 이미지에 대한 라벨링도 해주어야 해요.
라벨링 방법 라벨링은 객체 탐지에서 매우 중요한 과정이에요. YOLOv5에서는 LabelImg라는 툴을 사용하여 쉽게 라벨링할 수 있어요. LabelImg를 설치하고 실행한 후, 이미지를 불러온 뒤, 객체를 선택하고 라벨을 입력하면 끝이에요!
라벨링한 파일들은 YOLOv5의 format에 맞게 저장되어야 해요. 각 이미지와 같은 이름을 가진 .txt 파일이 생성되며, 해당 파일에는 객체의 클래스와 위치 정보가 담겨 있어야 해요.
모델 학습하기 이제 YOLOv5 모델을 학습시킬 차례예요. 학습을 위해서는 train.py 스크립트를 사용하면 돼요. 기본적인 학습 명령어는 다음과 같아요:
여기서 --source는 탐지할 이미지 파일을 의미해요. 이렇게 하면 탐지 결과가 저장된 폴더가 생성돼요.
결과 시각화 탐지 결과를 시각화하는 것은 매우 중요해요. YOLOv5는 탐지된 객체의 위치를 이미지에 표시해주며, 결과를 시각적으로 확인할 수 있어요. 결과 이미지에는 각 객체의 클래스와 신뢰도도 함께 나타나기 때문에, 얼마나 정확하게 탐지했는지를 쉽게 판단할 수 있답니다.
자주 묻는 질문 마지막으로, YOLOv5 사용 중에 자주 발생하는 문제와 그 해결 방법에 대해서도 알아볼게요:
모델 학습이 잘 되지 않는 경우 : 데이터셋이 불균형할 경우, 클래스의 개수를 조정하거나 더 많은 데이터를 추가하면 좋아요. 탐지 결과가 좋지 않은 경우 : 이미지 해상도나 모델 파라미터를 조정해보세요. 이상으로 YOLOv5의 사용 방법에 대해 알아보았어요. YOLOv5를 활용하여 다양한 프로젝트를 진행해보세요!
# define LED_PIN 5 // 핀번호 설정
void setup(){
pinMode(LED_PIN, OUTPUT); // 해당 핀을 출력으로 설정
}
void loop(){
for (int i = 0; i <= 255; i++){
analogWrite(LED_PIN, i); // LED의 밝기를 1씩 증가시켜서 최대값인 255까지 증가시키겠다.
delay(10);
}
for (int i = 255; i >= 0; i--){
analogWrite(LED_PIN, i); // LED의 밝기는 1씩 감소시켜 최소값이 0까지 감소시키겠다
delay(10);
}
}
/*
LED의 핀번호를 설정하고, 해당 핀을 출력으로 설정,
반복문을 통해 해당 핀의 LED 밝기를 0 > 255 로 1씩 증가시켜 점점 밝아지고,
255 > 0 으로 1씩 감소시켜 점점 어두워지는것을 확인 할 수 있음.
*/
EX_02>
#define LED_PIN 5 // LED 핀번호 설정
#define PLUS_BUTTON_PIN 34 // PLUS 핀번호 설정
#define MINUS_BUTTON_PIN 35 // MINUS 핀번호 설정
int count = 0; // 증가 값
void setup(){
pinMode(PLUS_BUTTON_PIN, INPUT_PULLUP); // LED의 해당 핀을 출력으로 설정
pinMode(MINUS_BUTTON_PIN, INPUT_PULLUP); // BUTTON의 해당 핀을 입력으로 설정
}
void loop(){
// 플러스 버튼이 눌렸냐?
if (digitalRead(PLUS_BUTTON_PIN) == LOW) {
// count를 1 증가대입
count = (count < 100) ? count + 1 : count;
delay(200);
}
// 마이너스 버튼이 눌렸냐?
if (digitalRead(MINUS_BUTTON_PIN) == LOW) {
// count를 1 감소대입
count = (count > 0) ? count - 1 : count;
delay(200);
}
// count의 0~50값을 0~255로 환산해서 LED의 밝기를 컨트롤
analogWrite(LED_PIN, map(count, 0, 50, 0, 255);
}
/*
버튼을 누를 때 마다 count 변수의 값이 상승, 하락하고
변수의 값은 최소 0 최대 50일때 map 함수를 통해 0~255로 환산해서 밝기를 조절 하는 코드
*/