changed data/loaders and engine/predictor so they accept 1 channel images for predictions

This commit is contained in:
Clea Parcerisas
2023-08-24 14:42:29 +02:00
parent 95e2c2424a
commit 82b97d5d4c
2 changed files with 3 additions and 2 deletions

View File

@ -249,7 +249,7 @@ class LoadImages:
else:
# Read image
self.count += 1
im0 = cv2.imread(path) # BGR
im0 = cv2.imread(path, cv2.IMREAD_GRAYSCALE) # BGR
if im0 is None:
raise FileNotFoundError(f'Image Not Found {path}')
s = f'image {self.count}/{self.nf} {path}: '