3.2 ResNet 이미지 분류기 데모
데모 환경
- Demo 환경
- Ubuntu 18.04
- GPU: Nvidia GTX 1080
- Data Platform: Anaconda
- 가상환경: pytorch_env2
- 가상환경: pytorch_env2
- Python: 3.7
- PyTorch: 1.0
ResNet을 이용한 이미지 분류기: 데이터 준비
ResNet 이미지 분류기를 학습시킬 데이터를 내려받고 지정된 디렉터리에 배치하는 동영상입니다. 데이터는 Kaggle 데이터를 사용합니다.
- 데이터 다운로드: https://www.kaggle.com/c/dogs-vs-cats
- 데이터 사이즈: 800Mb
ResNet을 이용한 이미지 분류기: 알고리즘 구현
- ResNet 알고리즘 학습
- 학습 주기: 24 epoch
- 학습 시간: 24min 32s (GTX 1080)