All Projects → chenyuntc → Pytorch Best Practice

chenyuntc / Pytorch Best Practice

A Guidance on PyTorch Coding Style Based on Kaggle Dogs vs. Cats

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to Pytorch Best Practice

Activation Visualization Histogram
Compare SELUs (scaled exponential linear units) with other activations on MNIST, CIFAR10, etc.
Stars: ✭ 377 (-35.78%)
Mutual labels:  image-classification
Caer
High-performance Vision library in Python. Scale your research, not boilerplate.
Stars: ✭ 452 (-23%)
Mutual labels:  image-classification
Fishnet
Implementation code of the paper: FishNet: A Versatile Backbone for Image, Region, and Pixel Level Prediction, NeurIPS 2018
Stars: ✭ 533 (-9.2%)
Mutual labels:  image-classification
Computer Vision
Programming Assignments and Lectures for Stanford's CS 231: Convolutional Neural Networks for Visual Recognition
Stars: ✭ 408 (-30.49%)
Mutual labels:  image-classification
Tfclassifier
Tensorflow based training and classification scripts for text, images, etc
Stars: ✭ 441 (-24.87%)
Mutual labels:  image-classification
Bag of tricks for image classification with convolutional neural networks
experiments on Paper <Bag of Tricks for Image Classification with Convolutional Neural Networks> and other useful tricks to improve CNN acc
Stars: ✭ 486 (-17.21%)
Mutual labels:  image-classification
Basic cnns tensorflow2
A tensorflow2 implementation of some basic CNNs(MobileNetV1/V2/V3, EfficientNet, ResNeXt, InceptionV4, InceptionResNetV1/V2, SENet, SqueezeNet, DenseNet, ShuffleNetV2, ResNet).
Stars: ✭ 374 (-36.29%)
Mutual labels:  image-classification
Dogs vs cats
猫狗大战
Stars: ✭ 570 (-2.9%)
Mutual labels:  image-classification
Seefood
Inspired by HBO's Silicon Valley: SeeFood is an iOS app that uses CoreML to detect various dishes
Stars: ✭ 445 (-24.19%)
Mutual labels:  image-classification
Residual Attention Network
Residual Attention Network for Image Classification
Stars: ✭ 525 (-10.56%)
Mutual labels:  image-classification
Bottleneck Transformer Pytorch
Implementation of Bottleneck Transformer in Pytorch
Stars: ✭ 408 (-30.49%)
Mutual labels:  image-classification
Neural Backed Decision Trees
Making decision trees competitive with neural networks on CIFAR10, CIFAR100, TinyImagenet200, Imagenet
Stars: ✭ 411 (-29.98%)
Mutual labels:  image-classification
Gluon Cv
Gluon CV Toolkit
Stars: ✭ 5,001 (+751.96%)
Mutual labels:  image-classification
Pytorch classification
利用pytorch实现图像分类的一个完整的代码,训练,预测,TTA,模型融合,模型部署,cnn提取特征,svm或者随机森林等进行分类,模型蒸馏,一个完整的代码
Stars: ✭ 395 (-32.71%)
Mutual labels:  image-classification
Mmclassification
OpenMMLab Image Classification Toolbox and Benchmark
Stars: ✭ 532 (-9.37%)
Mutual labels:  image-classification
Rmdl
RMDL: Random Multimodel Deep Learning for Classification
Stars: ✭ 375 (-36.12%)
Mutual labels:  image-classification
Pba
Efficient Learning of Augmentation Policy Schedules
Stars: ✭ 461 (-21.47%)
Mutual labels:  image-classification
Cvat
Powerful and efficient Computer Vision Annotation Tool (CVAT)
Stars: ✭ 6,557 (+1017.04%)
Mutual labels:  image-classification
Have Fun With Machine Learning
An absolute beginner's guide to Machine Learning and Image Classification with Neural Networks
Stars: ✭ 4,912 (+736.8%)
Mutual labels:  image-classification
Random Erasing
Random Erasing Data Augmentation. Experiments on CIFAR10, CIFAR100 and Fashion-MNIST
Stars: ✭ 511 (-12.95%)
Mutual labels:  image-classification

PyTorch 实践指南

本文是文章PyTorch实践指南配套代码,请参照知乎专栏原文或者对应的markdown文件更好的了解而文件组织和代码细节。

本部分内容属于 《深度学习框架PyTorch:入门与实践》一部分, 关于该书的源码,以及更多案例,请查看github

数据下载

  • kaggle比赛官网 下载所需的数据
  • 解压并把训练集和测试集分别放在一个文件夹中

安装

  • PyTorch : 可按照PyTorch官网的指南,根据自己的平台安装指定的版本
  • 安装指定依赖:
pip install -r requirements.txt

训练

必须首先启动visdom:

python -m visdom.server

然后使用如下命令启动训练:

# 在gpu0上训练,并把可视化结果保存在visdom 的classifier env上
python main.py train --data-root=./data/train --use-gpu=True --env=classifier

详细的使用命令 可使用

python main.py help

测试

python main.py --data-root=./data/test --use-gpu=False --batch-size=256
Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].