Skip to content

Commit 92e6f6a

Browse files
authored
Create README.md for image classification
1 parent 1312783 commit 92e6f6a

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

im-imageclassification/README.md

+5
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
### Iron Man Image classification full training
2+
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.
3+
4+
### Iron Man Image classification transfer learning
5+
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.

0 commit comments

Comments
 (0)