Method test_custom_image

ML/tests/test_model.py:149–174

(self, image_path)

        print(f"Overall Accuracy: {100.0 * overall_correct / overall_total:.2f}% ({overall_correct}/{overall_total})")
    
    def test_custom_image(self, image_path):
        if not os.path.exists(image_path):
            print(f"Error: Image not found at {image_path}")
            return
        
        try:
            image = Image.open(image_path).convert('RGB')
            
            transform = transforms.Compose([
                transforms.Resize((self.image_size, self.image_size)),
                transforms.ToTensor(),
            ])
            
            image_tensor = transform(image)
            pred_class, confidence, top5_idx, top5_prob = self.predict_image(image_tensor)
            
            print(f"\nCustom Image: {image_path}")
            print("-" * 60)
            print(f"Predicted:       {self.classes[pred_class]}")
            print(f"Confidence:      {confidence:.2%}")
            print("\nTop-5 Predictions:")
            for i, (idx, prob) in enumerate(zip(top5_idx, top5_prob), 1):
                print(f"  {i}. {self.classes[idx]:15s} {prob:.2%}")
        
        except Exception as e:
            print(f"Error loading image: {e}")
    
    def interactive_mode(self):
        print("\n" + "=" * 60)