ahadhassan commited on
Commit
2cd872e
·
verified ·
1 Parent(s): 6e63a4a

Update yolo_predictor.py

Browse files
Files changed (1) hide show
  1. yolo_predictor.py +1 -1
yolo_predictor.py CHANGED
@@ -130,7 +130,7 @@ def predict_pipeline(ndvi_model, yolo_model, rgb_array, conf=0.001):
130
  logger.info(f"NDVI prediction shape: {ndvi_prediction.shape}, range: [{ndvi_prediction.min():.3f}, {ndvi_prediction.max():.3f}]")
131
 
132
  # Step 3: Create 4-channel TIFF file
133
- logger.info("Step 3: Creating 4-channel TIFF file (BGR+NDVI)")
134
 
135
  # Create temporary file for the 4-channel TIFF
136
  with tempfile.NamedTemporaryFile(delete=False, suffix='.tiff') as tmp_file:
 
130
  logger.info(f"NDVI prediction shape: {ndvi_prediction.shape}, range: [{ndvi_prediction.min():.3f}, {ndvi_prediction.max():.3f}]")
131
 
132
  # Step 3: Create 4-channel TIFF file
133
+ logger.info("Step 3: Creating 4-channel TIFF file (RGB+NDVI)")
134
 
135
  # Create temporary file for the 4-channel TIFF
136
  with tempfile.NamedTemporaryFile(delete=False, suffix='.tiff') as tmp_file: