Overview

This repository implements classifier of cats and dogs for kaggle competition, "Dogs vs. Cats Redux: Kernels Edition". Resnet50 is used for transfer learning.

Link : https://www.kaggle.com/c/dogs-vs-cats-redux-kernels-edition/overview

Getting Started

The only thing you have to do is to execute "main.ipynb".

Environment

• CUDA 10.1
• Python 3.7
• NVIDIA GTX 1080 Ti

License

This project is licensed under MIT License - see the LICENSE file for details

References

• "PyTorch로 시작하는 딥러닝 - 딥러닝 기초에서 최신 모던 아키텍처까지", 에이콘출판사, 비슈누 수브라마니안 지음, 김태완 옮김