An open API service indexing awesome lists of open source software.

https://github.com/zeroact/intel_image_classfication


https://github.com/zeroact/intel_image_classfication

Last synced: 2 months ago
JSON representation

Awesome Lists containing this project

README

        

# Intel Image Scene Classification

kaggle의 Image Scene Classification of Multiclass 데이터셋을 사용하여 장면 분류 모델을 만듭니다.

### 라이브러리

python 3.7

tensorflow 2.2

matplotlib

json

### 데이터셋

https://www.kaggle.com/puneet6060/intel-image-classification

위 링크에서 데이터를 받아

dataset 디렉토리에 위치 시켜주세요.

### 파일 설명

test.py ~ test6.py는 성능이 좋은 하이퍼파라미터를 걸러내기 위한 실험 소스코드입니다.

최종모델은 final.py 한 소스코드 파일에 정리 되어 있습니다.

### 모델 구조

ResNet50 + Dense Layer + Dropout (0.5) + Dense Layer

세부 설명

https://lhj5682.blog.me/222015694137

### 모델 성능

![final_acc](final_acc.png)

![final_loss](final_loss.png)

max train accuracy : 95.97 %

max validation accuracy : 93.62 %

### ToDo

PyQt로 장면 분류기 만들기