Tag: pytorch
All the articles with the tag "pytorch".
-
Image Classification Training Pipeline — An 11-Step Checklist
I abstracted the training flow I learned yesterday through the NDT hands-on exercise into a step-by-step checklist applicable to any image classification problem. Along with small questions like why normalization is needed.
-
NDT Defect Classification Hands-on — From PyTorch + ResNet18 Transfer Learning to Grad-CAM
An opportunity came up to put my embedded object recognition experience to use, so I picked up NDT (non-destructive testing) again. A hands-on record of training a 6-class defect classifier on the NEU Steel Defect dataset using transfer learning, and visualizing the model's decision basis with Grad-CAM.