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

PyTorch Image Classification with Kaggle Dogs vs Cats Dataset

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

用户评价
全部评价

热门资源

  • Keras-ResNeXt

    Keras ResNeXt Implementation of ResNeXt models...

  • seetafaceJNI

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

  • spark-corenlp

    This package wraps Stanford CoreNLP annotators ...

  • capsnet-with-caps...

    CapsNet with capsule-wise convolution Project ...

  • inferno-boilerplate

    This is a very basic boilerplate example for pe...