deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Print error when image read fails

Browse Source
This commit is contained in:
James Betker 2020-04-23 23:59:32 -06:00
parent e98d92fc77
commit ea54c7618a
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/data/util.py
View File

@ -79,6 +79,8 @@ def read_img(env, path, size=None):
img = cv2.imread(path, cv2.IMREAD_UNCHANGED) img = cv2.imread(path, cv2.IMREAD_UNCHANGED)
else: else:
img = _read_img_lmdb(env, path, size) img = _read_img_lmdb(env, path, size)
if img is None:
print("Image error: %s" % (path,))
img = img.astype(np.float32) / 255. img = img.astype(np.float32) / 255.
if img.ndim == 2: if img.ndim == 2:
img = np.expand_dims(img, axis=2) img = np.expand_dims(img, axis=2)