ahadhassan commited on
Commit
29b059a
·
verified ·
1 Parent(s): 3bd81ab

Update yolo_predictor.py

Browse files
Files changed (1) hide show
  1. yolo_predictor.py +1 -1
yolo_predictor.py CHANGED
@@ -2,7 +2,7 @@
2
  import os
3
  import numpy as np
4
  import rasterio
5
- from ultralytics import YOLO
6
  from ndvi_predictor import normalize_rgb, predict_ndvi
7
  import tempfile
8
  from PIL import Image
 
2
  import os
3
  import numpy as np
4
  import rasterio
5
+ from modified_ultralytics import YOLO
6
  from ndvi_predictor import normalize_rgb, predict_ndvi
7
  import tempfile
8
  from PIL import Image