- Introduction - Training Image Classification Models
2
- Getting Started - Transfer Learning with PyTorch
3
- Example 1 - Re-training on Cat/Dog Dataset
4
- Example 1 - Step 1: Downloading the data
5
- Re-training ResNet-18 Model
6
- Exporting Pytorch model to Onyx and testing the model
7
- Testing on a live data stream
8
- Training on a custom dataset
9
- Testing the trained model on a live camera feed
10
- Conclusion
Description:
Dive into the world of image classification model training with PyTorch on Jetson Nano in this comprehensive 33-minute video. Learn transfer learning techniques, starting with re-training a ResNet-18 model on a cat/dog dataset. Follow step-by-step instructions for downloading data, re-training the model, and exporting it to ONNX format. Explore testing on live data streams and creating custom datasets for personalized models. Gain hands-on experience in applying trained models to live camera feeds, equipping you with practical skills for real-world AI applications on Jetson devices.
Jetson AI Fundamentals - Training Image Classification Models