Branch: master
Find file History
Pull request Compare This branch is 1 commit ahead, 12 commits behind awslabs:master.
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
..
Failed to load latest commit information.
Image-classification-fulltraining-elastic-inference.ipynb
Image-classification-fulltraining-highlevel-neo.ipynb
Image-classification-fulltraining-highlevel.ipynb
Image-classification-fulltraining.ipynb
Image-classification-incremental-training-highlevel.ipynb
Image-classification-lst-format-highlevel.ipynb
Image-classification-lst-format.ipynb
Image-classification-transfer-learning-highlevel.ipynb
Image-classification-transfer-learning.ipynb
README.md

README.md

SageMaker Image classification full training

This notebook ImageClassification-fulltraining.ipynb demos an end-2-end system for image classification training using resnet model. Caltech-256 dataset is used as a sample dataset. Various parameters such as network depth (number of layers), batch_size, learning_rate, etc., can be varied in the training. Once the training is complete, the notebook shows how to host the trained model for inference.

SageMaker Image classification transfer learning

This notebook Imageclassification-transfer-learning.ipynb demos an end-2-end system for image classification fine-tuning using a pre-trained resnet model on imagenet dataset. Caltech-256 dataset is used as a transfer learning dataset. The network re-initializes the output layer with the number of classes in the Caltech dataset and retrains the layer while at the same time fine-tuning the other layers. Various parameters such as network depth (number of layers), batch_size, learning_rate, etc., can be varied in the training. Once the training is complete, the notebook shows how to host the trained model for inference.

SageMaker Image classification lst format

This notebook Imageclassification-lst-format.ipynb demos an end-2-end system for image classification training with image and list files. Caltech-256 dataset is used as a transfer learning dataset. The network re-initializes the output layer with the number of classes in the Caltech dataset and retrains the layer while at the same time fine-tuning the other layers. Various parameters such as network depth (number of layers), batch_size, learning_rate, etc., can be varied in the training. Once the training is complete, the notebook shows how to host the trained model for inference.

SageMaker Image classification full training highlevel

This notebook ImageClassification-fulltraining-highlevel.ipynb is similar to the ImageClassification-fulltraining.ipynb but using Sagemaker high-level APIs

SageMaker Image classification full training highlevel with Neo

This notebook ImageClassification-fulltraining-highlevel-neo.ipynb is similar to the ImageClassification-fulltraining.ipynb but using Sagemaker high-level APIs(including Neo)

SageMaker Image classification transfer learning highlevel

This notebook Imageclassification-transfer-learning-highlevel.ipynb is similar to the ImageClassification-transfer-learning.ipynb but using Sagemaker high-level APIs

SageMaker Image classification lst format highlevel

This notebook Imageclassification-lst-format-highlevel.ipynb is similar to the ImageClassification-lst-format.ipynb but using Sagemaker high-level APIs