DawnC commited on
Commit
9ebccfe
·
verified ·
1 Parent(s): 7007d91

Update image_processor.py

Browse files
Files changed (1) hide show
  1. image_processor.py +2 -2
image_processor.py CHANGED
@@ -393,7 +393,7 @@ class ImageProcessor:
393
  if hasattr(self.scene_analyzer, 'spatial_analyzer') and self.scene_analyzer.spatial_analyzer:
394
  self.scene_analyzer.spatial_analyzer.update_class_names(current_class_names)
395
 
396
- logger.info(f"Updated class names in scene analyzer: {list(current_class_names.keys())}")
397
 
398
  self.scene_analyzer.enable_landmark = enable_landmark
399
  if hasattr(self.scene_analyzer, 'spatial_analyzer') and self.scene_analyzer.spatial_analyzer:
@@ -641,4 +641,4 @@ class ImageProcessor:
641
  # Descending order
642
  viz_data["class_data"].sort(key=lambda x: x["count"], reverse=True)
643
 
644
- return viz_data
 
393
  if hasattr(self.scene_analyzer, 'spatial_analyzer') and self.scene_analyzer.spatial_analyzer:
394
  self.scene_analyzer.spatial_analyzer.update_class_names(current_class_names)
395
 
396
+ self.logger.info(f"Updated class names in scene analyzer: {list(current_class_names.keys())}")
397
 
398
  self.scene_analyzer.enable_landmark = enable_landmark
399
  if hasattr(self.scene_analyzer, 'spatial_analyzer') and self.scene_analyzer.spatial_analyzer:
 
641
  # Descending order
642
  viz_data["class_data"].sort(key=lambda x: x["count"], reverse=True)
643
 
644
+ return viz_data