/cnn-dogs-vs-cats

Implementation of cats-vs-dogs based on CNN.

Primary LanguagePythonMIT LicenseMIT

cnn-dogs-vs-cats

Implementation of cats-vs-dogs based on CNN.

Environment

torch==1.10.1+cu111

torchvision==0.5.0

numpy==1.18.5

Usage

python CNN.py