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

PyTorch Image Classification with Kaggle Dogs vs Cats Dataset

2019-09-10 | |  212 |   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

用户评价
全部评价

热门资源

  • allennlp-server

    allennlp-server Serve allennlp services as sep...

  • allennlp_extras

    allennlp_extras Some utilities build on top of...

  • qp_testing

    qp_testing mkdir build cd build cmake .. make

  • allennlp-speech

    allennlp-speech

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...