데이터 취업 스쿨 스터디 노트 -(69) CNN(LeNET) 실습 - 마스크 여부 분별

테리·2024년 9월 9일
  • LeNET은 신경망 모델중 하나로 손으로 쓴 숫자를 인식하기 위해 개발된 초기의 컨볼루션 신경망(CNN). -> MNIST 데이터 분석에서 사용함.
  • 3개의 컨볼루션 레이어와 2개의 풀링 레이어 구조

실습

윈도우 환경에서 파이썬으로 파일 경로 보기
os.listdir()

필요한거 전부 import

import numpy as np
import pandas as pd
import os
import glob
import matplotlib.pyplot as plt
import seaborn as sns

import tensorflow as tf
from tensorflow.keras import Sequential, models
from tensorflow.keras.layers import Flatten, Dense, Conv2D, MaxPool2D
from sklearn.model_selection import train_test_split
from sklearn.metrics import classification_report, confusion_matrix 

경로와 목록 정리

  • OpenCV는 컴퓨터 비전과 이미지 처리를 위한 강력한 오픈소스 라이브러리
  • cv2는 OpenCV의 Python 모듈로, 이미지 및 비디오 처리와 컴퓨터 비전 관련 작업을 쉽게 수행할 수 있는 도구

어떤 그림들이 있는지 확인

데이터 나누고 잘 나눠졌는지 확인(데이터 나누는건 사실 안해도 됨. where에 이미 구분이 되어있어서)

인덱스 정리

데이터 전처리

shuffle()로 데이터 섞기

train, test 데이터 나누기

모델 만들기

compile 및 학습

validation accuracy가 나쁘지 않음.

예측

틀린거 찾아 표현하기

0개의 댓글