Add tutorial for torchvision

YeonwooSung · Jul 9, 2022 · db97326 · db97326
1 parent dc11097
commit db97326
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/MachineLearning/DeepLearning/Frameworks/PyTorch/README.md b/MachineLearning/DeepLearning/Frameworks/PyTorch/README.md
@@ -4,6 +4,10 @@
 
 ## Features
 
+## Related libraries
+
+1. [torchvision](./torchvision/)
+
 ## Discussions
 
 ### CUDA

diff --git a/MachineLearning/DeepLearning/Frameworks/PyTorch/torchvision/README.md b/MachineLearning/DeepLearning/Frameworks/PyTorch/torchvision/README.md
@@ -0,0 +1,9 @@
+# torchvision
+
+The torchvision package consists of popular datasets, model architectures, and common image transformations for computer vision.
+
+## Using pretrained models
+
+By using the torchvision package, we could easily load and use the pretrained vision models such as Resnet and Alexnet.
+
+[Example notebook for using pretrained models with torchvision](./src/Torchvision_tutorial.ipynb)
diff --git a/MachineLearning/DeepLearning/Frameworks/PyTorch/torchvision/src/Torchvision_tutorial.ipynb b/MachineLearning/DeepLearning/Frameworks/PyTorch/torchvision/src/Torchvision_tutorial.ipynb