CNN을 활용한 이미지 분류 모델 생성 프로그램

티처블 머신을 보니까 나도 비슷한 프로그램을 만들어 보고 싶어졌다.

 

그래서 언능 파이썬을 켜고 이미지 분류 모델을 만드는 프로그램을 개발해 보았따.

 

 

고양이라는 클래스를 추가하고 고양이 사진을 모아둔 폴더를 선택하여 고양이 이미지를 입력

강아지라는 클래스를 추가하고 강아지 사진을 모아둔 폴더를 선택하여

(더 추가하려면 클래스 추가버튼으로 추가)

데이터셋을 세팅한후에 만들기 버튼을 통해 이미지 분류 모델을 만들어 내는 프로그램을 만들었으며

저장된 모델은 "model.h5" 파일로 저장,

 

이렇게 생성된 모델은 추후에 모델을 사용하여 이미지를 분류하는 프로그램을 개발하여 테스트를 할예정