Skip to main content

Official TensorFlow ResNet classifier

Project description



Models
######

resnet-32-cifar10
#################

*ResNet-32 classifier for CIFAR-10*

Operations
==========

train
^^^^^

*Train the ResNet model on CIFAR-10 data*

Flags
-----

**batch-size**
*Number of images per batch (default is 128)*

**epochs**
*Number of epochs to train (default is 250)*

**epochs-per-eval**
*Number of epochs to run in between evaluations (default is 10)*

References
==========

- https://github.com/tensorflow/models/tree/v.1.6.0/official/resnet
- https://arxiv.org/pdf/1512.03385.pdf
- https://arxiv.org/pdf/1603.05027.pdf

resnet-imagenet
###############

*ResNet classifier for ImageNet*

Note, this model requires that the ImageNet dataset be available and prepared
as TFRrecords. Refer to https://git.io/vFQNF for instructions on preparing the
dataset.

Operations
==========

train
^^^^^

*Train the ResNet model on ImageNet data*

Flags
-----

**batch-size**
*Number of images per batch (default is 128)*

**data-dir**
*ImageNet data location (required)*

**epochs**
*Number of epochs to train (default is 250)*

**epochs-per-eval**
*Number of epochs to run in between evaluations (default is 10)*

**resnet-size**
*Size of the ResNet model to use (18, 34, 50, 101, 152, or 200) (default
is 50)

Choices:
18
34
50
101
152
200

*

References
==========

- https://github.com/tensorflow/models/tree/v.1.6.0/official/resnet
- https://arxiv.org/pdf/1512.03385.pdf
- https://arxiv.org/pdf/1603.05027.pdf


Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Filename, size & hash SHA256 hash help File type Python version Upload date
gpkg.tensorflow.resnet-0.4.0-py2.py3-none-any.whl (4.9 kB) Copy SHA256 hash SHA256 Wheel py2.py3 Jul 26, 2018

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page