fix bug

model.py

@@ -51,7 +51,54 @@ class TrafficSignDetector:
         
         self.thickness = config['inference']['thickness']
         self.classes = config['classes']
+        
+        # Print model information
+        self._print_model_info()
     
+    def _print_model_info(self):
+        """
+        Print detailed information about the loaded model.
+        """
+        print("\n" + "="*80)
+        print("MODEL INFORMATION")
+        print("="*80)
+        
+        # Basic model info
+        print(f"Model type: {type(self.model)}")
+        print(f"Model device: {self.model.device}")
+        print(f"Confidence threshold: {self.conf_threshold}")
+        print(f"Number of classes: {len(self.classes)}")
+        
+        # Model architecture
+        try:
+            print(f"\nModel architecture:")
+            print(f"  - Task: {self.model.task if hasattr(self.model, 'task') else 'Unknown'}")
+            print(f"  - Model type: {self.model.model.__class__.__name__ if hasattr(self.model, 'model') else 'Unknown'}")
+            
+            # Model parameters
+            if hasattr(self.model, 'model') and hasattr(self.model.model, 'parameters'):
+                total_params = sum(p.numel() for p in self.model.model.parameters())
+                trainable_params = sum(p.numel() for p in self.model.model.parameters() if p.requires_grad)
+                print(f"  - Total parameters: {total_params:,}")
+                print(f"  - Trainable parameters: {trainable_params:,}")
+        except Exception as e:
+            print(f"  - Could not retrieve architecture details: {e}")
+        
+        # Class information
+        print(f"\nClasses ({len(self.classes)} total):")
+        for i, cls in enumerate(self.classes):
+            print(f"  {i}: {cls}")
+        
+        # Try to get model summary
+        try:
+            if hasattr(self.model, 'info'):
+                print(f"\nModel summary:")
+                self.model.info()
+        except Exception as e:
+            print(f"Could not get model summary: {e}")
+        
+        print("="*80 + "\n")
+
     def _ensure_square(self, image, target_size=640):
         """
         Adjust image to square while maintaining aspect ratio.