资源分类
》 资源 》算法》PyTorch Image Classification with Kaggle Dogs vs Cats Dataset

PyTorch Image Classification with Kaggle Dogs vs Cats Dataset

2019-09-10 | |  179 |   0 |   0

PyTorch Image Classification

Classifies an image as containing either a dog or a cat (using Kaggle's public dataset), but could easily be extended to other image classification problems.

Dependencies:

  • PyTorch / Torchvision

  • Numpy

  • PIL

  • CUDA

Data

The data directory structure I used was:

  • project

    • test

    • dogs

    • cats

    • dogs

    • cats

    • data

    • train

    • validation

    • test

Performance

The result of the notebook in this repo produced a log loss score on Kaggle's hidden dataset of 0.04988 -- further gains can probably be achieved by creating an ensemble of classifiers using this approach.


上一篇：ppo_pytorch_cpp

下一篇：AEGeAN

用户评价
全部评价

热门资源

  • DuReader_QANet_BiDAF

    Machine Reading Comprehension on DuReader Usin...

  • ETD_cataloguing_a...

    ETD catalouging project using allennlp

  • honk-models

    Honk models google-speech-dataset.pt: model fo...

  • couleurs-des-van-...

    couleurs-des-van-gogh

  • allennlp_extras

    allennlp_extras Some utilities build on top of...