Spaces:
Sleeping
Sleeping
Update yolo_predictor.py
Browse files- 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 (
|
| 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:
|