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

PyTorch Image Classification with Kaggle Dogs vs Cats Dataset

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

用户评价
全部评价

热门资源

  • TensorFlow-Course

    This repository aims to provide simple and read...

  • seetafaceJNI

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

  • mxnet_VanillaCNN

    This is a mxnet implementation of the Vanilla C...

  • DuReader_QANet_BiDAF

    Machine Reading Comprehension on DuReader Usin...

  • Klukshu-Sockeye-...

    KLUKSHU SOCKEYE PROJECTS 2016 This repositor...