LGX95 / dogs_vs_cats

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Transfer learning with Resnet152 in Pytorch for the Kaggle competition Dogs vs. Cats Redux: Kernels Edition

Dependencies:

  • Python 3.6.3
  • Pytorch 0.3.0
  • torchvision 0.2.0
  • PIL 4.2.1
  • tqdm 4.19.5

The datasets folder like this:

  • datasets/cat_vs_dog/train/dog/dog.xxx.jpg
  • datasets/cat_vs_dog/train/cat/cat.xxx.jpg
  • datasets/cat_vs_dog/val/dog/dog.xxx.jpg
  • datasets/cat_vs_dog/val/cat/cat.xxx.jpg
  • datasets/cat_vs_dog/test/xxx.jpg

About


Languages

Language:Python 100.0%