Computer Vision, Deep Learning, Explainable Ai, Anomaly Detection, Editors Pick.
Original VGG-16 architecture VS custom one. Image by Author During training, the first 10 convolutional layers are frozen, we train only the classification head and the last 3 convolutional layers. That’s is because our dataset is too small to finetune the whole model. Loss is Cross-Entropy;